input='cg': nodes=species=Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, cart=Array([[ 4.63231 , 14.926445 , 7.378164 ],
[13.013201 , 6.1036897, 1.4445913],
[ 4.1904454, 2.7190654, 16.200918 ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):>, graph_i=Array([ 0, 0, 0, ..., 28, 28, 28], dtype=int32):>,), edges=to_jimage=Array([[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 1, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, -1],
...,
[ 0, 0, 0],
[ 0, 0, -1],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 1],
...,
[ 0, -1, 0],
[ 0, 0, 1],
[ 0, 0, 0]],
...,
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]]], dtype=int8):>, receiver=Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,), n_node=Array([232, 16, 14, 12, 11, 8, 8, 8, 8, 6, 6, 6, 6,
5, 4, 4, 4, 4, 4, 4, 4, 4, 2, 2, 2, 2,
1, 1, 636, 0, 0, 0], dtype=int32):>, padding_mask=Array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, False, False, False, False], dtype=bool):>, graph_data=dataset_id=Array([1201492, 1188146, 1104795, 571486, 1101786, 1183237, 1039119,
1184517, 570775, 1206390, 1094428, 570543, 1094243, 1224410,
977057, 1226864, 981387, 1183684, 817, 1067619, 1185068,
978280, 1183157, 3, 1186027, 1008733, 1056308, 11421,
0, 0, 0, 0], dtype=int32):>, abc=Array([[17.64551 , 17.64551 , 17.64551 ],
[ 5.314079 , 7.981866 , 13.585366 ],
[ 9.068563 , 10.770479 , 6.539788 ],
[ 4.030373 , 4.030373 , 17.434923 ],
[ 8.339144 , 8.339144 , 8.339144 ],
[ 7.174608 , 7.174608 , 5.329328 ],
[ 5.818448 , 5.818448 , 4.881939 ],
[ 6.569105 , 6.569105 , 5.321442 ],
[ 5.035579 , 6.520755 , 8.478259 ],
[ 9.256599 , 9.256599 , 40.654644 ],
[ 2.7972655, 2.7972655, 14.372677 ],
[ 5.9031425, 5.9031425, 5.9031425],
[ 8.704269 , 8.704269 , 5.420509 ],
[ 3.0599005, 3.0599005, 7.6981006],
[ 5.6382823, 5.6382823, 5.6382823],
[11.215207 , 11.215207 , 11.215207 ],
[ 4.759273 , 4.759273 , 4.759273 ],
[ 3.566622 , 3.566622 , 3.566622 ],
[ 4.759013 , 4.759013 , 4.759013 ],
[ 4.050405 , 4.050405 , 10.993359 ],
[ 5.42816 , 5.42816 , 5.42816 ],
[ 5.347009 , 5.347009 , 5.347009 ],
[ 3.742236 , 3.742242 , 6.189603 ],
[ 5.144408 , 5.144408 , 9.990392 ],
[ 3.2738857, 3.2738857, 3.616084 ],
[ 2.9911237, 2.9911237, 5.003386 ],
[ 3.3352723, 3.3352723, 3.3352723],
[ 3.4960268, 3.4960268, 3.4960268],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):>, angles_rad=Array([[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.4376192 , 1.9056848 , 1.023059 ],
[1.5707964 , 1.5707964 , 2.1006012 ],
[2.601387 , 2.284953 , 1.0326055 ],
[1.5707964 , 1.5707964 , 2.0943952 ],
[1.5707964 , 1.5707964 , 2.0943952 ],
[1.5707964 , 1.5707964 , 2.0943952 ],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 2.094395 ],
[1.0471976 , 1.0471976 , 1.0471976 ],
[1.5707964 , 1.5707964 , 2.7535915 ],
[1.769831 , 1.769831 , 1.5759323 ],
[2.408761 , 2.408761 , 1.0626415 ],
[0.30338755, 0.30338755, 0.30338755],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 1.5707964 ],
[1.5707964 , 1.5707964 , 2.0943954 ],
[1.0471976 , 1.0471976 , 1.0471976 ],
[2.3004804 , 2.3004804 , 1.2310284 ],
[1.5707964 , 1.5707964 , 2.094394 ],
[1.831197 , 1.831197 , 1.5707964 ],
[1.5707964 , 1.5707964 , 2.094395 ],
[1.5707964 , 1.5707964 , 2.094395 ],
[2.0780263 , 1.8892237 , 1.7733471 ],
[1.9106332 , 1.9106332 , 1.9106332 ],
[0. , 0. , 0. ],
[0. , 0. , 0. ],
[0. , 0. , 0. ],
[0. , 0. , 0. ]], dtype=float32):>, lat=Array([[[17.64551 , 0. , 0. ],
[ 0. , 17.64551 , 0. ],
[ 0. , 0. , 17.64551 ]],
[[ 5.314079, 0. , 0. ],
[ 0. , 7.981866, 0. ],
[ 0. , 0. , 13.585366]],
[[ 8.392959, 2.843163, -1.927046],
[ 2.694852, 10.404804, 0.69358 ],
[-1.110394, 0.759876, 6.399878]],
[[ 2.004346, -3.496642, 0. ],
[ 2.004346, 3.496642, 0. ],
[ 0. , 0. , 17.434923]],
[[-2.225138, 3.463605, 7.252139],
[ 2.225138, -3.463605, 7.252139],
[ 2.225138, 3.463605, -7.252139]],
[[ 3.587304, -6.213393, 0. ],
[ 3.587304, 6.213393, 0. ],
[ 0. , 0. , 5.329328]],
[[ 2.909224, -5.038924, 0. ],
[ 2.909224, 5.038924, 0. ],
[ 0. , 0. , 4.881939]],
[[ 3.284552, -5.689012, 0. ],
[ 3.284552, 5.689012, 0. ],
[ 0. , 0. , 5.321442]],
[[ 5.035579, 0. , 0. ],
[ 0. , 6.520755, 0. ],
[ 0. , 0. , 8.478259]],
[[ 9.256599, 0. , 0. ],
[ 0. , 9.256599, 0. ],
[ 0. , 0. , 40.654644]],
[[ 1.398633, -2.422503, 0. ],
[ 1.398633, 2.422503, 0. ],
[ 0. , 0. , 14.372677]],
[[ 0. , 4.174152, 4.174152],
[ 4.174152, 0. , 4.174152],
[ 4.174152, 4.174152, 0. ]],
[[ 1.67806 , -8.540985, 0. ],
[ 1.67806 , 8.540985, 0. ],
[ 0. , 0. , 5.420509]],
[[-2.158113, -1.533874, 1.533874],
[-2.158113, 1.533874, -1.533874],
[ 2.158113, -5.225098, -5.225098]],
[[-2.020036, 2.020036, 4.860981],
[ 2.020036, -2.020036, 4.860981],
[ 2.020036, 2.020036, -4.860981]],
[[11.086418, -1.69476 , 0. ],
[11.086418, 1.69476 , 0. ],
[10.827343, 0. , 2.923955]],
[[ 4.759273, 0. , 0. ],
[ 0. , 4.759273, 0. ],
[ 0. , 0. , 4.759273]],
[[ 3.566622, 0. , 0. ],
[ 0. , 3.566622, 0. ],
[ 0. , 0. , 3.566622]],
[[ 4.759013, 0. , 0. ],
[ 0. , 4.759013, 0. ],
[ 0. , 0. , 4.759013]],
[[ 2.025202, -3.507754, 0. ],
[ 2.025202, 3.507754, 0. ],
[ 0. , 0. , 10.993359]],
[[ 0. , 3.838289, 3.838289],
[ 3.838289, 0. , 3.838289],
[ 3.838289, 3.838289, 0. ]],
[[-2.183014, 2.183014, 4.365708],
[ 2.183014, -2.183014, 4.365708],
[ 2.183014, 2.183014, -4.365708]],
[[ 3.742236, 0. , 0. ],
[-1.871117, 3.240879, 0. ],
[ 0. , 0. , 6.189603]],
[[ 5.144408, 0. , 0. ],
[ 0. , 5.144408, 0. ],
[-2.572204, -2.572204, 9.304593]],
[[ 1.636943, -2.835268, 0. ],
[ 1.636943, 2.835268, 0. ],
[ 0. , 0. , 3.616084]],
[[-1.495562, -2.590389, 0. ],
[-1.495562, 2.590389, 0. ],
[ 0. , 0. , -5.003386]],
[[-1.69122 , 1.954661, 2.10787 ],
[ 1.69122 , -1.954661, 2.10787 ],
[ 1.69122 , 1.954661, -2.10787 ]],
[[-2.018432, 2.018432, 2.018432],
[ 2.018432, -2.018432, 2.018432],
[ 2.018432, 2.018432, -2.018432]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]]], dtype=float32):>,), target_data=e_form=Array([ -5.304165 , -4.064923 , -3.795275 , -3.9209325, -4.64022 ,
-2.2674756, -3.1913881, -2.6991951, -3.319136 , -1.8715571,
-1.2820146, -2.3706703, -3.1981266, -3.760566 , -7.064603 ,
-6.0091367, -1.0844846, -5.946206 , -3.0690575, -4.146882 ,
-4.1864815, -2.0256476, -5.599641 , -0.8074138, -7.056124 ,
-4.29189 , -3.9436038, -13.988523 , 0. , 0. ,
0. , 0. ], dtype=float32):>, force=Array([[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.],
...,
[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.]], dtype=float32):>, stress=Array([[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., -0.],
[ 0., 0., 0.],
[-0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[-0., 0., 0.],
[ 0., -0., 0.],
[ 0., 0., -0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[-0., -0., 0.],
[-0., 0., -0.],
[ 0., -0., -0.]],
[[-0., 0., 0.],
[ 0., -0., 0.],
[ 0., 0., -0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[-0., 0., 0.],
[ 0., -0., 0.],
[ 0., 0., -0.]],
[[ 0., 0., 0.],
[-0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[-0., -0., 0.]],
[[ 0., -0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[-0., -0., 0.],
[-0., 0., 0.],
[ 0., 0., -0.]],
[[-0., 0., 0.],
[ 0., -0., 0.],
[ 0., 0., -0.]],
[[-0., 0., 0.],
[ 0., -0., 0.],
[ 0., 0., -0.]],
[[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]],
[[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]],
[[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]],
[[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]]], dtype=float32):>,),),}, children='mace': input='vectors': Array([[[-1.2857466 , -0.6274977 , -1.9019718 ],
[ 0.10389948, 2.3928032 , -0.02432537],
[-1.1487083 , -0.86756516, 1.9172044 ],
...,
[ 2.351521 , 3.189148 , -0.04195452],
[ 0.01731491, -2.3332405 , 3.3903365 ],
[ 3.4649363 , -2.0678806 , 1.2120891 ]],
[[ 1.2857466 , -0.6274977 , 1.901972 ],
[-0.10389996, 2.3928027 , 0.02432549],
[ 1.1487074 , -0.86756516, -1.9172037 ],
...,
[-2.3515215 , 3.1891484 , 0.0419544 ],
[-0.01731491, -2.3332405 , -3.3903358 ],
[-3.4649363 , -2.0678802 , -1.2120894 ]],
[[ 1.2857466 , 0.6274979 , -1.9019709 ],
[-0.10389948, -2.3928027 , -0.02432442],
[ 1.1487079 , 0.8675654 , 1.9172039 ],
...,
[-2.3515215 , -3.1891468 , -0.04195404],
[-0.01731491, 2.3332403 , 3.390337 ],
[-3.4649363 , 2.0678804 , 1.2120895 ]],
...,
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]]], dtype=float32):>, 'node_species': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,}, children='node_embedding': input='node_species': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>,}, children='embed': input='inputs': Array([51, 51, 51, ..., -1, -1, -1], dtype=int32):>,}, params=#=5696, embedding=Array([[-0.05798482, -0.08304382, 0.12003322, ..., 0.07052852,
0.15142813, -0.0598093 ],
[-0.03456462, 0.01984368, -0.17332341, ..., -0.0481185 ,
0.22668917, 0.02687136],
[ 0.18563029, -0.11536985, -0.06535678, ..., 0.05797365,
-0.08284449, -0.17198712],
...,
[ 0.17062189, -0.08401794, 0.05306774, ..., 0.1194639 ,
-0.19122234, 0.09860158],
[ 0.1738508 , 0.23271398, -0.23452091, ..., 0.12326889,
0.08777846, 0.02295465],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,)>, output=Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,)>,}, output=64x0e
[[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
...
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]]key=0,): Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32), :>,)>, 'edge_embedding': input='edge_lengths': Array([[2.379999 , 2.3951814, 2.3974721, ..., 3.9625845, 4.1156645,
4.2132025],
[2.3799992, 2.395181 , 2.397471 , ..., 3.962585 , 4.115664 ,
4.2132025],
[2.379998 , 2.395181 , 2.3974717, ..., 3.962584 , 4.1156645,
4.2132025],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):>,}, children='basis': input='d': Array([[2.379999 , 2.3951814, 2.3974721, ..., 3.9625845, 4.1156645,
4.2132025],
[2.3799992, 2.395181 , 2.397471 , ..., 3.962585 , 4.115664 ,
4.2132025],
[2.379998 , 2.395181 , 2.3974717, ..., 3.962584 , 4.1156645,
4.2132025],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):>, 'r_max': <jax.Array([7.], dtype=float32)>:>,}, params=#=17, mu=Array([0. , 0.4666667, 0.9333334, 1.4000001, 1.8666668, 2.3333335,
2.8000002, 3.266667 , 3.7333336, 4.2000003, 4.666667 , 5.1333337,
5.6000004, 6.066667 , 6.533334 , 7. ], dtype=float32):>, sigma=<jax.Array([0.25], dtype=float32)>:>,)>, output=Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32):>,)>, 'envelope': input='x': Array([[2.379999 , 2.3951814, 2.3974721, ..., 3.9625845, 4.1156645,
4.2132025],
[2.3799992, 2.395181 , 2.397471 , ..., 3.962585 , 4.115664 ,
4.2132025],
[2.379998 , 2.395181 , 2.3974717, ..., 3.962584 , 4.1156645,
4.2132025],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):>, 'r_max': <jax.Array([7.], dtype=float32)>:>,}, output=Array([[1., 1., 1., ..., 1., 1., 1.],
[1., 1., 1., ..., 1., 1., 1.],
[1., 1., 1., ..., 1., 1., 1.],
...,
[1., 1., 1., ..., 1., 1., 1.],
[1., 1., 1., ..., 1., 1., 1.],
[1., 1., 1., ..., 1., 1., 1.]], dtype=float32):>,)>,}, params=#=1, rmax=<jax.Array([7.], dtype=float32)>:>,)>, output=16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>,)>, 'layer_0': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 64x0e
[[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
...
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]]key=0,): Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32), :>, 'node_species': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>, 'species_embed': Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,}, children='interaction': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 64x0e
[[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
...
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]]key=0,): Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32), :>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,}, children='linear_intro': input='weights_or_input': 64x0e
[[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
...
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]]key=0,): Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32), :>,}, params=#=4096, w[0,0] 64x0e,64x0e=Array([[-2.0628698 , -0.5036308 , 0.9884175 , ..., -3.7790492 ,
-0.38087052, 2.582744 ],
[-1.518289 , 1.141503 , -0.28400683, ..., -0.7970728 ,
0.07545985, -1.0202969 ],
[-1.939557 , 0.5432834 , 0.62171865, ..., 0.59222794,
0.13574058, -0.30513355],
...,
[ 1.8337163 , 0.4085469 , -1.1865293 , ..., 0.65343875,
0.44828436, -0.49455267],
[ 1.3934042 , 0.8345584 , 0.593422 , ..., -0.6222108 ,
0.9093682 , 0.7075241 ],
[-0.6080206 , 1.0741322 , 1.8177677 , ..., -1.6014078 ,
2.6109416 , -0.01555199]], dtype=float32):>,)>, output=64x0e
[[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
...
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]]key=0,): Array([[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
...,
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979],
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979],
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979]], dtype=float32), :>,)>, 'SevenNetConv_0': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 64x0e
[[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
[-0.05783462 0.15279612 0.1999597 ... 0.03095588 -0.12107898
0.01186812]
...
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]
[-0.03164478 -0.26114336 0.1989973 ... 0.02650559 0.02112022
0.11240979]]key=0,): Array([[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
[-0.05783462, 0.15279612, 0.1999597 , ..., 0.03095588,
-0.12107898, 0.01186812],
...,
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979],
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979],
[-0.03164478, -0.26114336, 0.1989973 , ..., 0.02650559,
0.02112022, 0.11240979]], dtype=float32), :>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,}, children='LazyInMLP_0': input='x': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>,}, children='Dense_0': input='inputs': Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32):>,}, params=#=3072, kernel=Array([[ 0.15018418, 0.1600938 , 0.04155254, ..., -0.07806948,
-0.14351869, -0.1139233 ],
[ 0.10056593, 0.15756494, 0.05010428, ..., 0.09964179,
-0.12858285, 0.03707083],
[-0.14641824, -0.05037894, 0.07878245, ..., 0.00162434,
0.16827556, -0.02695419],
...,
[ 0.096613 , -0.13295303, 0.16503717, ..., 0.12471856,
-0.06549358, 0.0675175 ],
[ 0.0422709 , -0.120007 , -0.0339473 , ..., -0.14121053,
-0.22076155, -0.01254457],
[ 0.01046582, 0.1581828 , 0.00369725, ..., -0.07496257,
-0.01728779, -0.00733899]], dtype=float32):>,)>, output=Array([[[ 0.11993331, -0.03900864, -0.00043423, ..., 0.08434734,
-0.12374312, 0.00931473],
[ 0.12261529, -0.03499188, -0.00694336, ..., 0.08373936,
-0.11574107, 0.01066817],
[ 0.1230339 , -0.0343315 , -0.00792077, ..., 0.08363546,
-0.1144636 , 0.01084997],
...,
[-0.05495894, 0.1624191 , -0.04307372, ..., 0.0649294 ,
-0.01594771, -0.10834614],
[-0.06300418, 0.12185571, -0.04649995, ..., 0.07934105,
-0.00311057, -0.13180344],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
[[ 0.11993336, -0.03900858, -0.00043434, ..., 0.08434732,
-0.123743 , 0.00931476],
[ 0.1226152 , -0.03499202, -0.00694316, ..., 0.08373938,
-0.11574133, 0.01066813],
[ 0.12303367, -0.03433185, -0.00792026, ..., 0.08363552,
-0.11446426, 0.01084988],
...,
[-0.05495898, 0.16241896, -0.04307376, ..., 0.06492946,
-0.01594766, -0.10834621],
[-0.06300419, 0.1218558 , -0.04649998, ..., 0.07934102,
-0.00311061, -0.13180338],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
[[ 0.11993315, -0.03900887, -0.00043383, ..., 0.08434737,
-0.12374359, 0.00931464],
[ 0.1226152 , -0.03499202, -0.00694316, ..., 0.08373938,
-0.11574133, 0.01066813],
[ 0.12303381, -0.03433164, -0.00792056, ..., 0.08363549,
-0.11446387, 0.01084993],
...,
[-0.05495889, 0.16241924, -0.04307368, ..., 0.06492935,
-0.01594776, -0.10834607],
[-0.06300418, 0.12185571, -0.04649995, ..., 0.07934105,
-0.00311057, -0.13180344],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
...,
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]],
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]],
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]]], dtype=float32):>,)>,}, output=192x0e
[[[ 0.11993331 -0.03900864 -0.00043423 ... 0.08434734 -0.12374312
0.00931473]
[ 0.12261529 -0.03499188 -0.00694336 ... 0.08373936 -0.11574107
0.01066817]
[ 0.1230339 -0.0343315 -0.00792077 ... 0.08363546 -0.1144636
0.01084997]
...
[-0.05495894 0.1624191 -0.04307372 ... 0.0649294 -0.01594771
-0.10834614]
[-0.06300418 0.12185571 -0.04649995 ... 0.07934105 -0.00311057
-0.13180344]
[-0.05472591 0.1116456 -0.03283517 ... 0.08249014 0.00354057
-0.14337058]]
[[ 0.11993336 -0.03900858 -0.00043434 ... 0.08434732 -0.123743
0.00931476]
[ 0.1226152 -0.03499202 -0.00694316 ... 0.08373938 -0.11574133
0.01066813]
[ 0.12303367 -0.03433185 -0.00792026 ... 0.08363552 -0.11446426
0.01084988]
...
[-0.05495898 0.16241896 -0.04307376 ... 0.06492946 -0.01594766
-0.10834621]
[-0.06300419 0.1218558 -0.04649998 ... 0.07934102 -0.00311061
-0.13180338]
[-0.05472591 0.1116456 -0.03283517 ... 0.08249014 0.00354057
-0.14337058]]
[[ 0.11993315 -0.03900887 -0.00043383 ... 0.08434737 -0.12374359
0.00931464]
[ 0.1226152 -0.03499202 -0.00694316 ... 0.08373938 -0.11574133
0.01066813]
[ 0.12303381 -0.03433164 -0.00792056 ... 0.08363549 -0.11446387
0.01084993]
...
[-0.05495889 0.16241924 -0.04307368 ... 0.06492935 -0.01594776
-0.10834607]
[-0.06300418 0.12185571 -0.04649995 ... 0.07934105 -0.00311057
-0.13180344]
[-0.05472591 0.1116456 -0.03283517 ... 0.08249014 0.00354057
-0.14337058]]
...
[[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
...
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]]
[[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
...
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]]
[[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
...
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]
[ 0.16765866 0.18764089 0.05040145 ... -0.06061758 -0.16587922
-0.10745643]]]key=0,): Array([[[ 0.11993331, -0.03900864, -0.00043423, ..., 0.08434734,
-0.12374312, 0.00931473],
[ 0.12261529, -0.03499188, -0.00694336, ..., 0.08373936,
-0.11574107, 0.01066817],
[ 0.1230339 , -0.0343315 , -0.00792077, ..., 0.08363546,
-0.1144636 , 0.01084997],
...,
[-0.05495894, 0.1624191 , -0.04307372, ..., 0.0649294 ,
-0.01594771, -0.10834614],
[-0.06300418, 0.12185571, -0.04649995, ..., 0.07934105,
-0.00311057, -0.13180344],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
[[ 0.11993336, -0.03900858, -0.00043434, ..., 0.08434732,
-0.123743 , 0.00931476],
[ 0.1226152 , -0.03499202, -0.00694316, ..., 0.08373938,
-0.11574133, 0.01066813],
[ 0.12303367, -0.03433185, -0.00792026, ..., 0.08363552,
-0.11446426, 0.01084988],
...,
[-0.05495898, 0.16241896, -0.04307376, ..., 0.06492946,
-0.01594766, -0.10834621],
[-0.06300419, 0.1218558 , -0.04649998, ..., 0.07934102,
-0.00311061, -0.13180338],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
[[ 0.11993315, -0.03900887, -0.00043383, ..., 0.08434737,
-0.12374359, 0.00931464],
[ 0.1226152 , -0.03499202, -0.00694316, ..., 0.08373938,
-0.11574133, 0.01066813],
[ 0.12303381, -0.03433164, -0.00792056, ..., 0.08363549,
-0.11446387, 0.01084993],
...,
[-0.05495889, 0.16241924, -0.04307368, ..., 0.06492935,
-0.01594776, -0.10834607],
[-0.06300418, 0.12185571, -0.04649995, ..., 0.07934105,
-0.00311057, -0.13180344],
[-0.05472591, 0.1116456 , -0.03283517, ..., 0.08249014,
0.00354057, -0.14337058]],
...,
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]],
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]],
[[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
...,
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643],
[ 0.16765866, 0.18764089, 0.05040145, ..., -0.06061758,
-0.16587922, -0.10745643]]], dtype=float32), :>,)>,}, output=64x0e+64x1e+64x2e
[[-1.6370075e-04 2.1333480e-02 -5.4670624e-03 ... 5.3513446e-05
2.2990990e-04 1.3581877e-05]
[-1.6369933e-04 2.1333480e-02 -5.4670526e-03 ... 5.3512944e-05
-2.2991026e-04 1.3581644e-05]
[-1.6370097e-04 2.1333488e-02 -5.4670484e-03 ... 5.3513522e-05
-2.2990983e-04 1.3582101e-05]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.6370075e-04, 2.1333480e-02, -5.4670624e-03, ...,
5.3513446e-05, 2.2990990e-04, 1.3581877e-05],
[-1.6369933e-04, 2.1333480e-02, -5.4670526e-03, ...,
5.3512944e-05, -2.2991026e-04, 1.3581644e-05],
[-1.6370097e-04, 2.1333488e-02, -5.4670484e-03, ...,
5.3513522e-05, -2.2990983e-04, 1.3582101e-05],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :>,)>, 'linear_outro': input='weights_or_input': 64x0e+64x1e+64x2e
[[-1.6370075e-04 2.1333480e-02 -5.4670624e-03 ... 5.3513446e-05
2.2990990e-04 1.3581877e-05]
[-1.6369933e-04 2.1333480e-02 -5.4670526e-03 ... 5.3512944e-05
-2.2991026e-04 1.3581644e-05]
[-1.6370097e-04 2.1333488e-02 -5.4670484e-03 ... 5.3513522e-05
-2.2990983e-04 1.3582101e-05]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.6370075e-04, 2.1333480e-02, -5.4670624e-03, ...,
5.3513446e-05, 2.2990990e-04, 1.3581877e-05],
[-1.6369933e-04, 2.1333480e-02, -5.4670526e-03, ...,
5.3512944e-05, -2.2991026e-04, 1.3581644e-05],
[-1.6370097e-04, 2.1333488e-02, -5.4670484e-03, ...,
5.3513522e-05, -2.2990983e-04, 1.3582101e-05],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :>,}, params=#=4096, w[0,0] 64x0e,42x0e=Array([[-0.05851466, -0.77724713, 0.31927708, ..., 1.3576921 ,
-0.556682 , 1.2009138 ],
[-0.19410008, 0.9568319 , 0.7947119 , ..., 1.1807404 ,
0.517647 , -2.1774006 ],
[-0.4475102 , 1.3353703 , 0.58434385, ..., -0.36388028,
-0.5137109 , -0.6175078 ],
...,
[ 1.0165759 , 1.6714705 , 0.7059146 , ..., -1.7501591 ,
1.2670155 , 0.20049101],
[ 0.28542688, 0.6939634 , -2.3232584 , ..., -0.16616781,
-2.9142556 , -2.217919 ],
[ 0.6932595 , 1.0191412 , -1.4806395 , ..., -0.15512154,
-0.8497269 , -1.1830466 ]], dtype=float32):>, w[1,1] 64x1e,14x1e=Array([[-1.59518883e-01, 1.41791570e+00, 2.21669644e-01,
-4.37419154e-02, -4.07786429e-01, 4.19104584e-02,
8.15000236e-01, -1.06575298e+00, 2.31332690e-01,
-8.67977679e-01, -2.43596837e-01, 1.37314987e+00,
-5.67853868e-01, 1.20926142e+00],
[ 4.18184608e-01, 9.88012910e-01, 4.79283929e-01,
4.05216813e-01, -4.53155339e-02, 3.04929286e-01,
-4.46530461e-01, -7.13473380e-01, 1.31436062e+00,
2.20264196e-01, 5.41667044e-01, 1.17608178e+00,
1.55635846e+00, -5.63044488e-01],
[ 2.27828217e+00, -1.19096234e-01, -1.25007868e+00,
2.27740467e-01, -1.08649099e+00, 1.33022428e+00,
6.26785934e-01, -4.07621503e-01, 1.69393003e+00,
-9.18190420e-01, -1.03266478e+00, 1.74743390e+00,
-1.47159457e-01, -2.22934440e-01],
[-7.19408840e-02, -9.65024412e-01, -6.41756356e-01,
-5.39019585e-01, -1.05908167e+00, 1.17141283e+00,
-8.43702316e-01, 6.23266935e-01, -1.40739465e+00,
-3.11620831e-01, -1.36519849e+00, 1.31101382e+00,
-1.36568034e+00, 4.00233507e-01],
[ 5.88025928e-01, 1.49504697e+00, -3.33192945e-02,
1.69268298e+00, -7.69107565e-02, -2.59293709e-02,
-1.65471530e+00, 3.29475671e-01, 1.25122988e+00,
-1.23136736e-01, 7.96665967e-01, 3.89983684e-01,
-7.09606349e-01, 1.51138830e+00],
[ 2.19460154e+00, 1.56129003e+00, 1.31001592e+00,
6.35921359e-02, 2.91403711e-01, -1.60269868e+00,
1.41060269e+00, 2.30047211e-01, 1.75895169e-01,
-1.62390459e+00, 1.15706229e+00, -1.32773876e+00,
-3.97600502e-01, -8.89060438e-01],
[-5.30009806e-01, 3.60730112e-01, 1.62346184e+00,
-1.06464362e+00, -1.08264184e+00, -1.29571998e+00,
-7.60659575e-01, 2.22292572e-01, 6.84585035e-01,
7.55619287e-01, 7.48718500e-01, 7.46250987e-01,
-9.37910452e-02, 3.64393860e-01],
[-1.29112840e-01, 1.32457066e+00, 5.08385479e-01,
1.45020112e-01, 5.37387133e-01, -2.00610375e+00,
5.76789975e-01, -9.72226799e-01, 3.96304876e-01,
-2.64844298e-01, 1.68858850e+00, -3.00725251e-01,
1.00796556e+00, 1.08232856e+00],
[ 6.27418160e-01, -5.57703495e-01, 5.05260944e-01,
-5.31034112e-01, -2.80631959e-01, 5.73693931e-01,
1.82067728e+00, 7.02271283e-01, 1.09990120e+00,
-7.72184134e-01, -1.69746768e+00, -1.13295031e+00,
-2.14488602e+00, 1.51665524e-01],
[-5.54692984e-01, -2.52606302e-01, -6.63607419e-01,
-6.24572411e-02, 1.56663167e+00, 2.45051950e-01,
1.29921484e+00, 1.67366230e+00, -1.22327462e-01,
-1.86858618e+00, 3.69350106e-01, 2.62834013e-01,
1.13924515e+00, 1.21686792e+00],
[-5.34396231e-01, -3.49607393e-02, 9.13910940e-03,
3.18263806e-02, -4.65545326e-01, 9.43098247e-01,
1.52805316e+00, 1.01767600e+00, -1.14559281e+00,
-8.60026896e-01, -5.84148169e-01, -1.18220651e+00,
5.74060738e-01, 2.57207662e-01],
[ 8.93370986e-01, -3.79539728e-01, -2.10544199e-01,
-1.78421450e+00, -1.95816517e-01, 1.30551231e+00,
-9.78464782e-01, 9.03600380e-02, -1.16986573e+00,
-4.74569321e-01, 1.17801487e+00, -9.98079851e-02,
9.02410269e-01, -1.16005051e+00],
[ 2.68199635e+00, -1.62732077e+00, -9.07512009e-01,
-2.67923456e-02, 9.14170682e-01, 2.59210229e-01,
-5.27186990e-01, -1.35804403e+00, -2.16185465e-01,
1.76343322e+00, -1.07868075e+00, -2.16450065e-01,
1.43215448e-01, -9.40346181e-01],
[ 3.59393597e-01, -1.16165960e+00, 6.91237509e-01,
-8.96016121e-01, 1.84420609e+00, -1.62266850e-01,
1.31136394e+00, -6.83795363e-02, 6.60903901e-02,
-8.65507007e-01, 9.96877372e-01, -5.17773688e-01,
9.89456952e-01, 7.62613177e-01],
[-2.25437269e-01, 3.20470273e-01, 1.35640287e+00,
-9.55064222e-02, 1.53520453e+00, -1.35338211e+00,
-2.21024260e-01, 1.20712876e+00, -4.05579329e-01,
-1.18690264e+00, -4.77740854e-01, -1.15239930e+00,
5.79453886e-01, 3.64091486e-01],
[ 4.25477028e-01, 4.81079936e-01, 5.01669943e-01,
6.96359634e-01, -1.12025690e+00, -3.97884548e-02,
2.42208973e-01, 4.49677110e-01, -1.02688277e+00,
5.28696477e-01, -1.16673386e+00, -9.62717891e-01,
-5.92334569e-01, 5.70299029e-01],
[ 5.86799443e-01, -7.21446633e-01, 7.38204002e-01,
8.66894424e-01, 8.82951260e-01, 5.07400222e-02,
-6.32285839e-03, -5.76142371e-01, -6.37560427e-01,
4.69085604e-01, -2.30725694e+00, -1.01172268e+00,
-9.90320921e-01, 5.31307645e-02],
[ 2.47569442e+00, -7.14210987e-01, 5.40753424e-01,
6.86744571e-01, -1.74699470e-01, 6.89243734e-01,
-4.08674479e-02, -4.45183247e-01, 1.84834969e+00,
1.54152185e-01, 4.17874157e-01, -5.16682267e-01,
1.47712126e-01, -9.91372943e-01],
[-5.53651489e-02, 6.26905620e-01, 1.06280816e+00,
2.67100304e-01, -8.69222522e-01, 2.67499179e-01,
-1.64249349e+00, -1.94458768e-01, -2.10027307e-01,
4.40979600e-01, 2.67733306e-01, -1.57145724e-01,
1.25143051e+00, -1.81606364e+00],
[ 9.76426125e-01, 6.68031693e-01, -1.04024374e+00,
-1.17045367e+00, -1.16735959e+00, -1.49027359e+00,
-6.79857075e-01, -7.48099446e-01, 1.32089186e+00,
1.40745151e+00, -1.22985864e+00, -1.54855943e+00,
-7.61636853e-01, -4.77871746e-01],
[ 4.24777716e-01, -1.56255138e+00, 1.12043238e+00,
-1.14868367e+00, -1.17755783e+00, 4.62594330e-02,
2.17900380e-01, -1.67058623e+00, -1.93805620e-01,
-9.77982432e-02, -2.18475834e-01, -2.26252079e+00,
-2.85863727e-01, 7.70356953e-01],
[ 2.68083382e-02, 5.09472132e-01, 1.13345599e+00,
2.11207658e-01, -4.62850183e-01, -4.13339317e-01,
-6.48412183e-02, 6.58049524e-01, 1.01331997e+00,
-5.04264235e-01, 8.78735900e-01, 4.85170364e-01,
-8.16698492e-01, -2.30840063e+00],
[ 2.38270164e-02, -9.57201123e-01, 1.11666358e+00,
-1.30178726e+00, -1.37858719e-01, 5.67627311e-01,
-3.39662254e-01, -8.66294324e-01, 3.46518874e-01,
-3.37120473e-01, -1.19312108e+00, 1.70416367e+00,
-2.66984969e-01, -5.59353650e-01],
[ 5.48908830e-01, 1.55050850e+00, -6.24530792e-01,
1.68726087e+00, 4.32809830e-01, -5.27117670e-01,
-1.24971770e-01, 5.66418290e-01, 1.72672772e+00,
1.36358750e+00, -4.42286581e-01, 3.13295841e-01,
-1.30482769e+00, -1.05473423e+00],
[ 2.35792041e-01, -6.08000457e-01, -1.75224841e+00,
-5.61265588e-01, -5.46101034e-01, 9.14718509e-01,
5.02703786e-01, -9.03682053e-01, 2.34616414e-01,
-4.15714771e-01, 1.83502769e+00, -2.95173436e-01,
1.35221028e+00, 5.88385403e-01],
[-4.55738634e-01, 4.45710421e-01, -1.21132219e+00,
1.86995089e-01, -2.49144268e+00, 4.72594619e-01,
1.53070942e-01, 2.33280730e+00, 2.02912673e-01,
6.70928836e-01, -8.74447823e-01, 4.44694519e-01,
-2.07142636e-01, -1.55804050e+00],
[ 2.67416179e-01, 1.04169309e+00, -4.14637953e-01,
1.68740976e+00, 9.02660966e-01, 6.07355118e-01,
-4.00293559e-01, -7.39988804e-01, 1.58039913e-01,
-2.56460458e-01, 4.43774372e-01, 2.92048812e-01,
1.82465422e+00, -4.14018303e-01],
[ 1.52441597e+00, 4.89026070e-01, 1.21373260e+00,
4.35728312e-01, -1.07791650e+00, -2.45261049e+00,
9.35544431e-01, -1.35772705e+00, 1.79711056e+00,
8.52342367e-01, 1.19373155e+00, -6.13982260e-01,
5.70725203e-01, -9.68867719e-01],
[ 2.03021406e-03, -1.29999354e-01, 5.92655316e-02,
1.25821471e+00, 6.24942183e-02, 1.80472732e+00,
-1.16673493e+00, -5.93848765e-01, 4.71933186e-01,
-7.81845581e-03, -2.29606166e-01, 2.03171635e+00,
1.79288650e+00, 1.29030132e+00],
[ 6.08261943e-01, -1.42228925e+00, -5.33825517e-01,
3.43213826e-01, -6.70951545e-01, 4.96039679e-03,
-2.30120987e-01, -4.27372962e-01, 1.62094891e+00,
-1.28940022e+00, 1.39186287e+00, 1.51078928e+00,
6.47550464e-01, 1.98174155e+00],
[-6.12567030e-02, 8.57409000e-01, 1.32825351e+00,
-1.02852702e+00, 3.35335404e-01, -1.02163625e+00,
-4.15726528e-02, 1.93185598e-01, -4.74221230e-01,
-1.67110413e-01, 5.20935953e-01, -1.93280554e+00,
2.00355351e-01, -3.23708951e-01],
[ 1.08565259e+00, 1.61983907e-01, 1.71757948e+00,
3.40961248e-01, 1.04746819e+00, -2.30317369e-01,
-1.85366511e-01, -7.92828500e-01, 9.94528115e-01,
1.76955611e-01, 1.76184416e+00, 5.97553134e-01,
1.67578131e-01, 9.35976803e-01],
[-8.12537134e-01, 1.44042718e+00, 8.62172365e-01,
8.57314393e-02, 9.90101099e-01, -8.86001170e-01,
7.09160388e-01, -9.29489970e-01, 5.78041434e-01,
1.55850577e+00, -2.95356419e-02, -1.34474203e-01,
-4.29579943e-01, 3.91554713e-01],
[ 1.65328681e+00, -6.20100260e-01, -1.67706501e+00,
5.59364378e-01, -4.37381119e-01, -6.49949729e-01,
1.71196625e-01, -1.13286567e+00, 6.59414470e-01,
8.06130171e-02, -9.95175302e-01, 8.50464821e-01,
-3.74371737e-01, -1.02404332e+00],
[ 1.26137173e+00, -1.23269951e+00, 3.81778210e-01,
2.19730902e+00, 1.33422828e+00, 2.94234842e-01,
1.15006697e+00, -6.75675154e-01, 9.26281393e-01,
4.50897038e-01, -2.21485281e+00, 2.83924848e-01,
-9.56719816e-01, 8.49782467e-01],
[ 1.33975387e+00, 2.62724042e+00, -5.24466097e-01,
-7.74450362e-01, -9.57646668e-01, 1.56592774e+00,
-7.83164620e-01, -6.40024006e-01, -4.88792539e-01,
-1.06006289e+00, 1.44969714e+00, 4.93297368e-01,
-3.08405340e-01, -6.86704218e-01],
[-3.83207612e-02, -6.77714050e-01, -3.40886027e-01,
-1.13979447e+00, 4.87399161e-01, 7.72835553e-01,
-1.40748417e+00, -7.39935815e-01, 1.59411454e+00,
1.15674841e+00, -9.19692755e-01, 8.76317501e-01,
1.00377285e+00, 3.21376026e-01],
[ 5.06586075e-01, 1.21491992e+00, -9.25272107e-01,
1.06484696e-01, -8.16683531e-01, 5.93299747e-01,
-1.10479629e+00, 7.03715831e-02, -3.33572663e-02,
-4.64212626e-01, 2.35317722e-01, -8.59455764e-01,
1.22262096e+00, 1.37142134e+00],
[-1.38571036e+00, -3.22411537e-01, -6.36700571e-01,
2.69516140e-01, -8.14863026e-01, 1.28528440e+00,
1.76175952e+00, -6.53155625e-01, -6.37266755e-01,
-1.02621281e+00, -1.08890641e+00, 7.49036819e-02,
1.22699112e-01, 1.25357735e+00],
[ 4.52807158e-01, -8.89224112e-01, 4.98062402e-01,
1.89246997e-01, 1.31713465e-01, -5.23973465e-01,
1.54224753e-01, -6.78794980e-01, 1.14550853e+00,
1.51655555e+00, 3.64522576e-01, -1.17647633e-01,
-6.54954195e-01, 8.90015304e-01],
[-1.31320572e+00, -1.20855129e+00, -4.92994845e-01,
1.55690646e+00, 6.84804201e-01, -1.54884088e+00,
-7.45994210e-01, 4.18577969e-01, 2.22844720e+00,
-4.07182164e-02, -1.23469245e+00, -5.48134804e-01,
1.08503067e+00, -1.40429690e-01],
[ 1.50631130e+00, 5.24843156e-01, -1.82353187e+00,
1.87019721e-01, 6.12977564e-01, 1.11283529e+00,
6.42728925e-01, -2.18589020e+00, -2.42828593e-01,
-1.21111441e+00, 3.66430581e-01, 1.38284033e-02,
3.85612130e-01, 4.68372285e-01],
[-1.75874382e-01, -4.08292055e-01, -1.00944924e+00,
5.58892012e-01, -4.72226292e-01, -3.02028596e-01,
1.00368285e+00, -5.84220409e-01, -9.46995318e-01,
-6.55231595e-01, -4.84680831e-01, 1.36366165e+00,
3.50298882e-01, 6.15626693e-01],
[ 2.60027337e+00, 1.35086524e+00, -1.35592818e+00,
-1.58175099e+00, 3.57953221e-01, -7.68379509e-01,
4.03920293e-01, 1.67774391e+00, -4.20965552e-01,
-1.12308192e+00, -1.27487123e-01, -4.15252775e-01,
-1.22445464e-01, -4.65124696e-01],
[-3.16420585e-01, -6.81420743e-01, -1.26265013e+00,
-1.98174238e+00, 2.75979429e-01, -9.23054963e-02,
4.52505916e-01, 2.84654468e-01, 1.15839040e+00,
-7.42772758e-01, -8.08473885e-01, 1.08693194e+00,
1.71116877e+00, 8.35318387e-01],
[-1.30527645e-01, 1.19810545e+00, 4.04397875e-01,
1.05148375e+00, 1.16548412e-01, -4.16003048e-01,
7.92871237e-01, 3.64823371e-01, 1.34322092e-01,
-1.02372932e+00, -3.20769936e-01, 1.08710289e+00,
2.50287127e+00, 3.08796883e+00],
[ 1.35144436e+00, -2.24318361e+00, 1.41327202e+00,
-3.39333087e-01, 3.89240593e-01, 8.71712744e-01,
-5.47430038e-01, 2.60426730e-01, 5.06789267e-01,
2.15743613e+00, 5.25659800e-01, 1.16867685e+00,
5.52198470e-01, 1.89584062e-01],
[-1.41914308e+00, -1.80148780e+00, 1.03770339e+00,
-6.49519444e-01, -1.90983617e+00, -5.67432940e-01,
-5.59940457e-01, -5.88242412e-01, -1.78088024e-01,
-2.81739794e-02, 3.50285798e-01, -2.98163891e-01,
-9.50639009e-01, 1.19947755e+00],
[ 6.89429104e-01, -1.87503946e+00, 5.58116257e-01,
-6.97072923e-01, 1.58970022e+00, 2.63239235e-01,
-9.97781694e-01, 1.59912694e+00, -4.75035936e-01,
1.34154546e+00, 4.36315536e-01, 2.21418627e-02,
-6.78245783e-01, -2.59008551e+00],
[-8.97034049e-01, 2.22304523e-01, 6.84527680e-02,
7.21424699e-01, 1.68601289e-01, -1.46354496e-01,
-1.52812648e+00, -2.26916170e+00, 5.64270914e-01,
2.18850896e-01, 1.44817483e+00, 1.25659943e+00,
-4.17729259e-01, -2.16195869e+00],
[-1.61611950e+00, -1.76749599e+00, -4.22315031e-01,
-4.91583757e-02, -1.50362003e+00, 1.06815231e+00,
-9.07922685e-01, -4.16147679e-01, 1.72047985e+00,
2.30066836e-01, 8.54514718e-01, 3.22850198e-01,
2.05851769e+00, -1.07616329e+00],
[-2.58867711e-01, 3.74838591e-01, -1.77129745e+00,
6.04138613e-01, -8.48847985e-01, 1.12743664e+00,
-1.46299446e+00, 2.79084891e-01, -1.34576201e+00,
-6.64600551e-01, -2.20910370e-01, -1.71081436e+00,
9.06275153e-01, -6.68873608e-01],
[ 9.84097600e-01, 9.09519970e-01, -9.70170557e-01,
-1.44275323e-01, -4.49816108e-01, -1.88520384e+00,
3.29373777e-01, -5.58341034e-02, -4.68398780e-01,
9.11014080e-01, -9.09228802e-01, 2.64525384e-01,
-2.02725148e+00, 6.50482774e-02],
[ 8.36299419e-01, 1.30899537e+00, -5.31107068e-01,
-1.40431952e+00, -4.71691221e-01, -3.16469610e-01,
5.58074355e-01, 1.00490844e+00, -1.50469983e+00,
2.00777507e+00, -1.01735735e+00, 1.07961118e+00,
-6.60374999e-01, 6.59434915e-01],
[-7.27077186e-01, -1.03576732e+00, -1.48827061e-01,
-1.32349908e+00, -1.10893822e+00, 1.46627808e+00,
7.22141206e-01, -5.66078961e-01, -5.60155094e-01,
9.86186028e-01, 1.53977501e+00, -8.08483779e-01,
1.00651050e+00, 1.40823647e-01],
[-5.33250511e-01, 2.09856009e+00, -1.16714168e+00,
5.54437041e-02, -4.81565595e-01, -5.36132812e-01,
-2.76243901e+00, 1.53763723e+00, 3.11925918e-01,
2.72354987e-02, 9.56707746e-02, -3.35781932e-01,
-1.06018817e+00, -9.29266751e-01],
[ 1.21636200e+00, 8.46770465e-01, -1.69799075e-01,
1.74350667e+00, -1.47740114e+00, 1.81421793e+00,
2.81792134e-02, -7.02113867e-01, 9.75015342e-01,
8.97165179e-01, -1.38118660e+00, -1.15525818e+00,
5.07358432e-01, -9.23286438e-01],
[ 1.89580214e+00, -6.20792866e-01, -9.25718784e-01,
9.79650244e-02, 4.18433160e-01, -5.57581306e-01,
-1.26872456e+00, -1.14570570e+00, -8.84682536e-01,
-5.93434989e-01, -2.48874421e-03, 1.40595579e+00,
-4.25415039e-01, -1.03271830e+00],
[ 1.36196673e+00, 1.50914681e+00, -2.13788912e-01,
1.61337662e+00, 9.92569566e-01, 1.82063317e+00,
1.54667366e+00, 5.59173584e-01, -3.80773515e-01,
1.81888595e-01, -2.56054729e-01, -2.99480915e-01,
1.03786467e-02, -4.63755041e-01],
[ 1.84127986e-02, -6.76998854e-01, -4.29344326e-01,
1.90196231e-01, -2.68527269e-02, -1.52830625e+00,
1.68638206e+00, 4.11245048e-01, 1.26944029e+00,
-2.65661389e-01, -3.86289895e-01, 1.10908544e+00,
-1.35901928e+00, 6.57546043e-01],
[ 9.78038847e-01, 1.47112799e+00, -7.70605743e-01,
2.63111448e+00, -9.32955682e-01, -1.81793702e+00,
5.46021104e-01, -8.80700707e-01, 5.71719646e-01,
-9.51708615e-01, 6.31415248e-02, 6.01736069e-01,
2.36932069e-01, -9.86322880e-01],
[ 5.31408370e-01, 9.14077640e-01, -1.07504475e+00,
-2.18011737e+00, -7.92915642e-01, 1.03825033e+00,
-6.16012692e-01, 8.11390996e-01, 6.76774740e-01,
-2.41251633e-01, 7.86914706e-01, -1.00438043e-01,
1.44513583e+00, 4.69384164e-01],
[-8.99046957e-01, 7.23022282e-01, -7.45066643e-01,
4.92994010e-01, -5.81285357e-02, -8.28124464e-01,
3.93308103e-01, -2.07284912e-01, -2.20448947e+00,
-2.45271951e-01, 3.84791166e-01, 3.15079838e-02,
-1.09610856e+00, 2.20290199e-01],
[-5.08466184e-01, 3.31172794e-01, -5.46600580e-01,
-1.51968569e-01, -1.82997179e+00, 7.28380308e-02,
-7.87612617e-01, -2.44463414e-01, 1.65684617e+00,
-2.40155369e-01, 2.21262053e-01, 8.79563570e-01,
-4.44383293e-01, 2.34226465e+00]], dtype=float32):>, w[2,2] 64x2e,8x2e=Array([[ 9.67481956e-02, 6.48988724e-01, -1.57308489e-01,
1.04333878e-01, 7.60372519e-01, -1.12586968e-01,
1.21055432e-01, 7.37992823e-01],
[-7.74415314e-01, -5.29010773e-01, 6.47630453e-01,
1.68535614e+00, -4.34721291e-01, 3.32736641e-01,
-1.16851091e+00, -1.77664235e-01],
[ 7.03454435e-01, 5.85008979e-01, -3.61304358e-02,
-1.22629857e+00, -1.40319812e+00, -2.09124714e-01,
-4.68867332e-01, -8.87821168e-02],
[-2.24138212e+00, 5.01526892e-01, 1.43474936e+00,
-1.20937884e-01, 1.16449308e+00, -8.28431770e-02,
5.07275522e-01, -8.32767665e-01],
[ 7.30961442e-01, 1.94547403e+00, -4.46520895e-01,
1.63447344e+00, -8.09260130e-01, 4.05759871e-01,
-5.64258814e-01, 1.36827064e+00],
[ 2.36636430e-01, -1.23017120e+00, 2.22999573e+00,
-6.41391039e-01, 8.06245327e-01, 5.21986902e-01,
-5.48370957e-01, 9.24639285e-01],
[-7.08031058e-01, 2.16012096e+00, -1.03522432e+00,
-1.28307569e+00, -9.04882789e-01, -7.16227710e-01,
-1.32459962e+00, -9.88584682e-02],
[ 5.31849921e-01, 2.72150666e-01, 1.38804579e+00,
-6.71439230e-01, -1.09824741e+00, 1.22396791e+00,
5.28674461e-02, 5.52391648e-01],
[-1.49875975e+00, -6.74942061e-02, 1.07143867e+00,
6.48121774e-01, 2.73639178e+00, -1.03295469e+00,
1.26032782e+00, 2.61963665e-01],
[-8.57270241e-01, -6.99343085e-01, 8.67248178e-01,
-4.74980287e-02, 8.35231602e-01, 2.85025388e-01,
2.43002415e+00, 2.34641563e-02],
[ 9.43550527e-01, 1.30916643e+00, -1.11989093e+00,
-1.66024470e+00, 1.83724433e-01, 8.72271180e-01,
1.24853730e+00, -8.79910946e-01],
[ 1.05873084e+00, -1.49080265e+00, 3.86952430e-01,
6.97257668e-02, 6.81466937e-01, 8.10701370e-01,
3.96183997e-01, 1.66515782e-01],
[ 1.10028875e+00, 1.71403217e+00, -1.23441339e+00,
-8.41688573e-01, -2.35026643e-01, -4.06826288e-02,
-7.70527244e-01, -2.10039213e-01],
[ 8.04340303e-01, -7.22950697e-01, 1.86933076e+00,
1.03510869e+00, -1.15992896e-01, 1.67192087e-01,
2.72174239e-01, 7.24298120e-01],
[-8.57638359e-01, -2.69152552e-01, 1.89254451e+00,
1.81740057e-02, -1.31002188e+00, -2.04619989e-02,
-1.47427750e+00, 1.05544460e+00],
[ 5.16046405e-01, -9.62329268e-01, 5.96567512e-01,
1.88853666e-01, -3.67044091e-01, 1.14778328e+00,
-2.40221202e-01, -2.29378536e-01],
[-4.93779182e-01, 4.77483749e-01, -9.87859309e-01,
7.44566441e-01, -4.33188707e-01, 4.69831258e-01,
-1.05279958e+00, -1.06508888e-01],
[-7.65590549e-01, -2.47405291e-01, 5.44655681e-01,
-1.72484457e+00, -2.46677566e+00, 3.41684341e-01,
-5.16951442e-01, 6.13680124e-01],
[-6.12389445e-01, 1.44171312e-01, 1.17558742e+00,
5.83570637e-02, -5.69417655e-01, -4.73689318e-01,
-1.05864727e+00, -1.32746184e+00],
[ 1.28330410e+00, 6.56110108e-01, -2.08690739e+00,
2.13454485e+00, -1.21887326e-01, -1.17592168e+00,
-6.21322930e-01, -1.99168175e-01],
[-2.72104949e-01, -1.07604325e-01, -9.07125115e-01,
4.56590295e-01, -5.67522168e-01, 2.11338019e+00,
1.78962362e+00, 1.09979463e+00],
[-2.41228580e-01, 1.02694738e+00, -1.13448799e+00,
1.09150696e+00, 6.04097009e-01, -2.06960216e-02,
1.90655395e-01, -8.22154999e-01],
[-4.95553493e-01, -5.04372180e-01, -8.37219417e-01,
-2.13859558e-01, 9.33445990e-01, -5.53758860e-01,
1.10436320e+00, 2.19170499e+00],
[ 2.83110499e-01, -5.06958365e-01, 1.05745041e+00,
-6.56313419e-01, -3.84220868e-01, 9.74942327e-01,
3.33282769e-01, -5.34421243e-02],
[ 2.54492462e-01, 1.84631014e+00, -8.63231599e-01,
1.85738444e-01, 5.36234081e-01, -8.65710914e-01,
3.55212092e-02, -5.78615963e-01],
[ 6.24077082e-01, -1.20511305e+00, -7.96263099e-01,
9.48948503e-01, 4.31757361e-01, -1.84780967e+00,
3.50582898e-01, -1.31270480e+00],
[-1.61597955e+00, 2.53971481e+00, -2.51417089e+00,
-3.74935865e-01, -1.31955087e+00, 2.21978337e-01,
-9.10206795e-01, 9.64830339e-01],
[-9.34218943e-01, 7.05695331e-01, 2.03276253e+00,
-1.52681696e+00, 1.39523292e+00, -1.37838364e+00,
8.04154038e-01, -3.43164921e+00],
[-9.99716997e-01, -3.24213445e-01, -1.34170306e+00,
-2.04816863e-01, -1.07242084e+00, -9.61314797e-01,
-4.18930382e-01, 1.13090360e+00],
[ 4.80788440e-01, -5.55313110e-01, -1.15892684e+00,
-1.15143120e+00, 5.54329693e-01, -8.59109461e-01,
3.95437777e-01, -6.51472509e-02],
[ 6.34740770e-01, -1.51537442e+00, -1.60719857e-01,
-1.68346837e-01, 9.59250391e-01, -9.48030353e-01,
-2.96364665e-01, -1.79281998e+00],
[ 2.70316321e-02, 1.05667257e+00, -3.58083069e-01,
-2.20257610e-01, 1.12218440e+00, -7.62031853e-01,
6.17300093e-01, -9.96770561e-01],
[-5.43457448e-01, 7.14095414e-01, -6.10465050e-01,
-1.63760915e-01, 5.95864654e-01, -1.49819028e+00,
5.11981547e-01, 1.15930185e-01],
[-4.34810936e-01, 9.44709599e-01, -1.10076201e+00,
1.87155771e+00, 2.21008825e+00, -6.13352060e-01,
-1.36835325e+00, -1.12006831e+00],
[-7.00573564e-01, 4.96258914e-01, -1.86095071e+00,
-4.51034456e-01, -1.59216547e+00, 4.35990989e-02,
-9.49206114e-01, 2.95604944e-01],
[ 2.93732494e-01, 2.01401472e-01, 2.01494485e-01,
-1.25133669e+00, -1.64028645e-01, -1.16593218e+00,
-3.32203150e-01, 8.90385807e-01],
[ 1.23134685e+00, -6.55903935e-01, -1.17262340e+00,
6.13659978e-01, 5.97331040e-02, -6.90859854e-01,
-1.18477976e+00, 2.41937578e-01],
[-4.99329567e-02, -8.22783709e-01, -1.98960245e+00,
-7.14252591e-01, 1.18851125e-01, 9.07236159e-01,
-1.62274420e-01, 4.15660471e-01],
[ 1.85519886e+00, -1.62114632e+00, -5.12873769e-01,
-1.49721634e+00, 2.10612208e-01, -1.46814942e+00,
1.31215855e-01, -2.08199069e-01],
[ 6.87979817e-01, -7.12743253e-02, -2.21066207e-01,
7.84528732e-01, -3.69821221e-01, -6.29335865e-02,
-2.37908363e+00, -6.00247860e-01],
[ 6.38839006e-01, -7.47517347e-01, -3.14254642e-01,
1.38885513e-01, -1.18265676e+00, -8.07747781e-01,
7.13936567e-01, -6.70344949e-01],
[ 4.40051734e-01, -1.04402229e-01, 1.13665295e+00,
-8.56030822e-01, 2.01950386e-01, 3.28441083e-01,
-6.59650639e-02, -1.18293665e-01],
[-4.13568765e-01, 4.42352653e-01, 3.79088134e-01,
6.78174138e-01, -8.30674827e-01, 6.80784881e-01,
2.21875444e-01, -3.47600281e-01],
[-7.43939936e-01, 3.30214292e-01, -3.06804442e+00,
3.17766122e-03, -1.32356358e+00, 9.09575164e-01,
-1.71026206e+00, -9.46841002e-01],
[-4.86533672e-01, 1.77362275e+00, -1.02714419e+00,
8.78805876e-01, 1.50514352e+00, 8.28731954e-01,
1.61510423e-01, -1.23058975e+00],
[ 3.45375866e-01, 3.12800646e-01, -1.44201291e+00,
-4.22390491e-01, 9.24703002e-01, -1.49287239e-01,
-2.76055038e-01, -5.52495241e-01],
[-6.40695691e-02, -2.10809097e-01, -5.08558035e-01,
-1.07067537e+00, -4.32312220e-01, 1.56954169e+00,
8.52030396e-01, 5.94061017e-01],
[ 9.69568193e-01, -9.55774188e-01, -2.31819317e-01,
-1.17909348e+00, 7.90362433e-02, 4.48527839e-03,
1.15838802e+00, 3.43258500e-01],
[-1.01521444e+00, 1.40422142e+00, 3.56133103e-01,
9.62435484e-01, -1.54768318e-01, 1.16666389e+00,
-6.72136068e-01, -3.71779442e-01],
[ 8.52488041e-01, -2.99089044e-01, 7.79101402e-02,
-5.31521380e-01, 1.61081898e+00, 4.16966856e-01,
1.26738620e+00, 6.67820275e-01],
[ 9.90517318e-01, 3.99273671e-02, -2.85845995e+00,
4.14837718e-01, -1.11525536e+00, 2.26733994e+00,
6.05014980e-01, -1.71796834e+00],
[ 4.18335289e-01, 2.17945671e+00, 1.06952012e-01,
9.02975321e-01, -1.54626989e+00, 4.30744022e-01,
4.04606074e-01, 1.43895343e-01],
[ 5.66147864e-01, -7.43111223e-02, 7.53086627e-01,
2.39888978e+00, 1.18667209e+00, 6.90562189e-01,
-6.97662473e-01, 6.03483200e-01],
[-7.07864404e-01, 4.97037679e-01, -1.06388199e+00,
-8.95456493e-01, 1.77361977e+00, -3.33036870e-01,
5.88657916e-01, 1.44794226e-01],
[ 9.70811486e-01, -5.10534763e-01, 1.78254735e+00,
-3.00022840e+00, 1.48504317e+00, 8.38298425e-02,
-9.82980251e-01, -1.20963252e+00],
[-2.45066285e-01, 6.34577051e-02, -7.76195049e-01,
2.94479787e-01, 1.66743505e+00, -1.26691031e+00,
-4.73201960e-01, 4.16079491e-01],
[ 1.15633070e+00, -1.21601677e+00, 1.66564143e+00,
-9.57550764e-01, -1.44023371e+00, 2.87680030e-01,
9.22170699e-01, -1.86783671e+00],
[ 5.08022368e-01, 1.82837105e+00, -9.91158038e-02,
-1.17294681e+00, 2.89817780e-01, -8.83939341e-02,
1.42732129e-01, -2.68379307e+00],
[-3.15176487e-01, 1.50880590e-01, -9.78354871e-01,
-1.22338951e+00, 8.93037498e-01, 8.31120253e-01,
3.97685654e-02, 2.91188270e-01],
[ 3.82317901e-01, -2.13419333e-01, -1.52989328e+00,
-4.09932494e-01, 9.31123078e-01, 1.03175497e+00,
-2.35781026e+00, 8.37871432e-01],
[-1.83633339e+00, 7.10269809e-01, -1.80358243e+00,
1.22166121e+00, 6.57803893e-01, -8.67673695e-01,
-1.52248487e-01, -1.11806750e+00],
[ 4.96033132e-01, -9.27934945e-01, 1.24262226e+00,
3.75043780e-01, -1.72185516e+00, 2.73734003e-01,
-7.53707170e-01, 9.29422855e-01],
[ 8.33779812e-01, 3.95064324e-01, 9.03998613e-01,
5.73015332e-01, -1.58878481e+00, -1.76471722e+00,
2.59873748e-01, 2.04126492e-01],
[-1.12557256e+00, -1.79209161e+00, -3.34454089e-01,
-1.09964275e+00, 1.61668515e+00, -1.66957760e+00,
-1.07278235e-01, -2.78059989e-02]], dtype=float32):>,)>, output=42x0e+14x1e+8x2e
[[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021414 0.00688458
0.00278721]
[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021415 -0.00688458
0.00278723]
[ 0.00789474 -0.00415786 -0.00684164 ... 0.00021415 -0.00688458
0.00278721]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021414,
0.00688458, 0.00278721],
[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021415,
-0.00688458, 0.00278723],
[ 0.00789474, -0.00415786, -0.00684164, ..., 0.00021415,
-0.00688458, 0.00278721],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>,}, output=42x0e+14x1e+8x2e
[[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021414 0.00688458
0.00278721]
[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021415 -0.00688458
0.00278723]
[ 0.00789474 -0.00415786 -0.00684164 ... 0.00021415 -0.00688458
0.00278721]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021414,
0.00688458, 0.00278721],
[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021415,
-0.00688458, 0.00278723],
[ 0.00789474, -0.00415786, -0.00684164, ..., 0.00021415,
-0.00688458, 0.00278721],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'self_connection': input='node_feats': 42x0e+14x1e+8x2e
[[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021414 0.00688458
0.00278721]
[ 0.00789475 -0.00415786 -0.00684165 ... 0.00021415 -0.00688458
0.00278723]
[ 0.00789474 -0.00415786 -0.00684164 ... 0.00021415 -0.00688458
0.00278721]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021414,
0.00688458, 0.00278721],
[ 0.00789475, -0.00415786, -0.00684165, ..., 0.00021415,
-0.00688458, 0.00278723],
[ 0.00789474, -0.00415786, -0.00684164, ..., 0.00021415,
-0.00688458, 0.00278721],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>, 'node_specie': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, 'species_embed': Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,}, children='self_gate': input='x': Array([[ 6.2327133e-05, -3.2825268e-05, -5.4013148e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
[ 6.2327148e-05, -3.2825290e-05, -5.4013122e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
[ 6.2326930e-05, -3.2825199e-05, -5.4012969e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02]], dtype=float32):>,}, children='Dense_0': input='inputs': Array([[ 6.2327133e-05, -3.2825268e-05, -5.4013148e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
[ 6.2327148e-05, -3.2825290e-05, -5.4013122e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
[ 6.2326930e-05, -3.2825199e-05, -5.4012969e-05, ...,
1.6483489e-03, -3.8677990e-02, -9.4289377e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
-2.2383080e-01, -3.0349290e-01, -8.7981731e-02]], dtype=float32):>,}, params=#=45936, kernel=Array([[ 0.04216176, -0.03840219, 0.06802341, ..., 0.0085216 ,
0.00269195, -0.0224273 ],
[-0.00129787, -0.04396937, -0.00605627, ..., -0.0540319 ,
-0.01580891, -0.04374993],
[ 0.02092985, 0.05780528, 0.05336943, ..., 0.02976701,
0.01025869, 0.01234752],
...,
[-0.03091729, -0.00359123, -0.05720197, ..., 0.02794363,
-0.03799417, 0.02786122],
[-0.03145697, -0.01792482, -0.03916818, ..., -0.0562315 ,
0.01010597, -0.05003702],
[-0.03580622, 0.02172906, 0.02199378, ..., 0.04462478,
-0.06405319, 0.0304339 ]], dtype=float32):>,)>, output=Array([[ 0.01908433, -0.02076847, 0.02098764, ..., -0.0345698 ,
0.02400569, -0.01213535],
[ 0.01908433, -0.02076847, 0.02098764, ..., -0.0345698 ,
0.02400569, -0.01213535],
[ 0.01908433, -0.02076847, 0.02098764, ..., -0.0345698 ,
0.02400569, -0.01213535],
...,
[ 0.04752952, -0.00486182, 0.03504989, ..., -0.00636473,
0.03760951, 0.01513482],
[ 0.04752952, -0.00486182, 0.03504989, ..., -0.00636473,
0.03760951, 0.01513482],
[ 0.04752952, -0.00486182, 0.03504989, ..., -0.00636473,
0.03760951, 0.01513482]], dtype=float32):>,)>, 'Dropout_0': input='inputs': Array([[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
...,
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468]], dtype=float32):>,}, output=Array([[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
...,
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468]], dtype=float32):>,)>, 'LayerNorm_0': input='x': Array([[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
[ 0.00963321, -0.01027641, 0.01060393, ..., -0.01698616,
0.01214691, -0.00603086],
...,
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468],
[ 0.02432942, -0.002425 , 0.01783204, ..., -0.00317224,
0.01915833, 0.00762468]], dtype=float32):>,}, params=#=22, scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1.], dtype=float32):>,)>, output=Array([[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.9920583 , -0.27958524],
[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.99205834, -0.27958524],
[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.9920583 , -0.27958524],
...,
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ],
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ],
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ]], dtype=float32):>,)>, 'Dense_1': input='inputs': Array([[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.9920583 , -0.27958524],
[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.99205834, -0.27958524],
[ 0.8162104 , -0.57658666, 0.8841182 , ..., -1.0459744 ,
0.9920583 , -0.27958524],
...,
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ],
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ],
[ 1.5167437 , -0.10071962, 1.1239384 , ..., -0.14589474,
1.2041206 , 0.5068428 ]], dtype=float32):>,}, params=#=484, kernel=Array([[-6.26387000e-02, -4.05864954e-01, 6.65039644e-02,
1.39925241e-01, 1.26208998e-02, 2.62338847e-01,
1.48871213e-01, -5.48392870e-02, -1.52895093e-01,
-9.23760608e-02, 1.76248565e-01, 1.72931001e-01,
1.78557724e-01, -2.08691537e-01, 1.78452477e-01,
4.05558705e-01, 1.75475374e-01, -1.81462780e-01,
-1.40967518e-01, -2.96664476e-01, -5.71025684e-02,
-7.56553710e-02],
[ 3.01190853e-01, -4.26264666e-02, 3.73568051e-02,
1.81464598e-01, 1.73494250e-01, 1.09101109e-01,
-8.36101398e-02, -1.81374684e-01, -2.41980046e-01,
4.39208001e-01, -2.92042375e-01, -2.54162163e-01,
2.03091279e-01, 6.15993468e-03, -1.36905834e-01,
1.99727386e-01, 2.05471311e-02, -4.30944078e-02,
-2.21278276e-02, -1.02557220e-01, 2.05812216e-01,
-1.11809745e-01],
[ 1.53448889e-02, -4.82997335e-02, 6.93193600e-02,
-2.98031896e-01, -1.81696638e-01, -1.12288989e-01,
-1.40338510e-01, 4.62142862e-02, 1.87174231e-01,
-3.39084685e-01, 5.19301184e-02, 5.10093085e-02,
1.95269808e-01, -1.87210217e-02, -2.11240783e-01,
-4.09309305e-02, -1.88481659e-01, 6.07440770e-02,
-2.99154192e-01, 3.43036950e-01, -1.18522838e-01,
1.30719379e-01],
[-1.89546291e-02, -1.49124071e-01, 1.79182202e-01,
-9.24292952e-02, 2.29432449e-01, -1.90507434e-02,
-1.30989086e-02, -4.25095744e-02, 2.44076341e-01,
-3.51524681e-01, 2.75491923e-01, -1.89514354e-01,
1.68697447e-01, -1.84086710e-01, 1.41118377e-01,
3.00346941e-01, 1.45123258e-01, 2.29019672e-02,
-4.84605245e-02, -3.93821806e-01, 1.17301002e-01,
2.08854735e-01],
[ 2.58403867e-01, 1.66586667e-01, 3.85116309e-01,
-9.33973268e-02, 1.23489179e-01, 2.02582791e-01,
6.45986050e-02, -3.00178200e-01, -1.80001646e-01,
3.11329126e-01, -2.44925350e-01, -1.43404394e-01,
-7.01924190e-02, 1.09919906e-01, 2.55484670e-01,
3.53565037e-01, -1.39134556e-01, 3.96913476e-02,
-2.34727506e-02, -9.83282402e-02, 1.35175228e-01,
-9.09328684e-02],
[-1.20718785e-01, -2.08415240e-02, -1.64374843e-01,
-2.16466621e-01, -2.40984955e-03, -4.34296541e-02,
-3.30051780e-01, 3.94039601e-01, 3.82609218e-01,
2.20852017e-01, -4.76281673e-01, 3.17855626e-01,
-2.43194140e-02, -2.91409343e-01, -1.19395182e-01,
-1.90602411e-02, 6.27645329e-02, -3.60631526e-01,
-1.88289329e-01, -2.13994652e-01, -4.49951828e-01,
-1.03070870e-01],
[ 3.88505101e-01, -5.71238287e-02, 1.46077856e-01,
1.72224969e-01, -8.98153409e-02, -8.67100134e-02,
3.72179866e-01, 2.31264472e-01, -1.80253059e-01,
1.57859787e-01, -1.49951681e-01, -1.09648772e-01,
-3.98628235e-01, -4.39249068e-01, -1.25172153e-01,
2.08052725e-01, 2.84637272e-01, 5.27398437e-02,
1.61246791e-01, 2.96399407e-02, -9.53676701e-02,
-6.55202493e-02],
[ 1.24878742e-01, -2.48728350e-01, -2.42345817e-02,
5.93937896e-02, 2.07798049e-01, -3.62451404e-01,
3.11028250e-02, 3.61922011e-02, -3.93688381e-01,
3.73115897e-01, -2.24199906e-01, -1.79090545e-01,
2.45331049e-01, 3.23778570e-01, -6.25772215e-03,
3.89803320e-01, 3.25194538e-01, 1.12955362e-01,
-2.17728093e-01, 2.42776722e-02, -2.64965415e-01,
-1.67991623e-01],
[-1.84036255e-01, 2.70671099e-02, 1.75936762e-02,
-1.07529946e-01, 3.44970137e-01, -4.11004812e-01,
-8.63340423e-02, -2.20303476e-01, -4.40734532e-03,
7.78180137e-02, -4.22156276e-03, -3.86881948e-01,
3.89209151e-01, -8.50830823e-02, 9.18821916e-02,
7.48913959e-02, -1.40549794e-01, 4.05104637e-01,
1.95633933e-01, -1.44673616e-01, -6.05336614e-02,
-1.45517349e-01],
[ 7.34342728e-03, -1.05782278e-01, -1.40532866e-01,
-1.39351368e-01, 6.15580007e-02, -3.32787246e-01,
-2.83225775e-01, 5.11239357e-02, -1.11494452e-01,
3.25462073e-01, 2.52609879e-01, 2.58215331e-02,
-4.26878259e-02, 2.19850421e-01, -1.08119823e-01,
2.57674694e-01, 1.75447106e-01, -7.79689848e-02,
5.26501518e-03, -8.77768397e-02, 1.58996247e-02,
1.93164870e-02],
[-2.50227958e-01, 4.89520505e-02, -9.59638730e-02,
2.94916481e-01, -7.71423280e-02, 2.34870404e-01,
2.25777015e-01, 4.00733590e-01, 2.30387613e-01,
4.37932074e-01, 1.45891771e-01, 3.15020770e-01,
-2.90467948e-01, 1.61787838e-01, 4.55440283e-01,
-1.73544571e-01, 1.98901966e-02, 2.18838211e-02,
-1.23214476e-01, 2.54757017e-01, -7.97718018e-02,
1.45427883e-01],
[ 1.54784694e-01, -1.44105077e-01, -2.49068141e-01,
-7.68315652e-03, 1.42058551e-01, -1.17592543e-01,
1.43037498e-01, 3.11496288e-01, 2.54283577e-01,
1.11053847e-01, -1.32418782e-01, 4.58562374e-01,
1.40863582e-01, -1.26921698e-01, 8.44412073e-02,
-1.17169507e-01, 3.84026378e-01, 9.07778069e-02,
4.06687349e-01, -2.90428400e-01, 4.86307498e-03,
-3.07883978e-01],
[ 1.14953732e-02, -2.06889421e-01, -1.81866482e-01,
9.90195498e-02, 1.61663294e-01, -2.95546085e-01,
-1.47396758e-01, 2.35425130e-01, 7.66063556e-02,
1.77511781e-01, -2.57493168e-01, -4.19574119e-02,
1.12826854e-01, -2.67672151e-01, 5.61202653e-02,
1.11290753e-01, 3.57788987e-02, -7.44150877e-02,
-2.24923164e-01, -2.88073987e-01, -1.50604576e-01,
2.92692035e-01],
[-3.18740122e-02, 4.80743349e-02, 4.97874320e-02,
-3.58266205e-01, 2.78039515e-01, -3.94127488e-01,
3.22612096e-03, -4.44587730e-02, -4.55204807e-02,
2.08180636e-01, -4.00409162e-01, 1.26749203e-01,
-1.52946547e-01, 4.14981157e-01, -1.13116287e-01,
1.49217397e-01, 1.75830796e-01, 4.25507613e-02,
1.17730573e-01, -5.40154874e-02, 2.51689047e-01,
-1.49811745e-01],
[ 3.98961455e-01, -1.87527865e-01, 2.07536697e-01,
1.80399820e-01, -2.23493397e-01, 8.33448246e-02,
-2.96423197e-01, 1.08473554e-01, -2.64511526e-01,
-2.97478795e-01, -2.69568682e-01, 3.81647825e-01,
2.47910008e-01, 1.06414139e-01, 4.69201654e-02,
4.99183983e-02, 2.10724741e-01, -1.20360173e-01,
-7.53049627e-02, 2.61868834e-01, 3.99036612e-03,
8.76427069e-02],
[ 1.13861546e-01, 7.02895075e-02, -2.79774398e-01,
8.56420919e-02, -9.60764661e-02, 1.74400344e-01,
2.31235415e-01, 4.56745923e-02, 9.79280025e-02,
4.56151217e-01, -2.16886193e-01, 1.58790126e-01,
2.53539115e-01, 2.90547460e-02, -3.97028886e-02,
3.58660407e-02, -1.96615100e-01, 7.01555982e-02,
-1.23695292e-01, -5.82575016e-02, -3.62884589e-02,
-1.09403692e-01],
[-3.83595414e-02, 2.00851753e-01, -2.04384308e-02,
-1.68715507e-01, 2.91162450e-03, -2.57057458e-01,
-1.81814834e-01, -1.94544613e-01, 5.95297031e-02,
-1.45526037e-01, 2.28887662e-01, -2.10195929e-01,
-1.15945868e-01, -2.33043972e-02, 1.18407227e-01,
1.59461573e-01, 8.53893608e-02, -2.83727944e-01,
1.05260648e-02, 4.26474512e-01, 2.53956288e-01,
6.08125776e-02],
[ 2.58080631e-01, -1.77756652e-01, 2.75313139e-01,
-1.34941235e-01, 2.65602022e-04, -3.78166109e-01,
-5.55976555e-02, -1.30554706e-01, -1.01425417e-01,
-1.58725694e-01, 1.73681438e-01, 8.02538246e-02,
-4.30461437e-01, -4.46689337e-01, -1.79116622e-01,
1.36527985e-01, 2.06859961e-01, -2.34158337e-01,
3.22615087e-01, 1.31683454e-01, -2.64532566e-01,
2.87408143e-01],
[-3.88725102e-01, 1.69502478e-02, -5.66641279e-02,
1.82202701e-02, -4.15405720e-01, 2.65897244e-01,
-1.92075059e-01, 1.97343320e-01, 1.94640055e-01,
2.28250667e-01, 4.12942469e-01, -1.67736113e-01,
-1.10270865e-01, 1.47957459e-01, 8.28340948e-02,
-2.70363808e-01, 2.21281826e-01, 1.09487601e-01,
-1.48043297e-02, 1.44729197e-01, 7.46383071e-02,
-1.64278373e-02],
[-2.67697364e-01, 2.06662089e-01, -3.12386483e-01,
1.22382298e-01, -1.60692170e-01, 1.13184825e-01,
1.77272841e-01, 2.18634337e-01, -1.76084608e-01,
1.20963253e-01, -2.19279841e-01, 3.53890389e-01,
-1.71338186e-01, 1.25111893e-01, -4.10987854e-01,
-1.24395669e-01, -2.90392518e-01, -2.76591539e-01,
-1.04178116e-01, -1.06795788e-01, 3.15787201e-03,
1.00351453e-01],
[-3.29942793e-01, -2.49078885e-01, -1.48973316e-01,
-3.51872742e-02, 2.80188113e-01, -1.60527200e-01,
2.31993616e-01, 3.41932386e-01, 9.24585983e-02,
-7.19041675e-02, -2.55450457e-01, 1.08463958e-01,
1.86314747e-01, -1.57371953e-01, -3.58703732e-02,
8.47460628e-02, -3.08688492e-01, 1.39280185e-02,
-2.69706428e-01, 1.29534870e-01, -1.48895666e-01,
-6.94105998e-02],
[ 3.37186232e-02, -1.86767191e-01, 1.94881428e-02,
1.36405855e-01, 3.02507758e-01, 2.91096181e-01,
-2.94693381e-01, 2.71930277e-01, -1.25490531e-01,
8.00061822e-02, -7.67261088e-02, 4.71446693e-01,
-8.45387727e-02, 1.68739989e-01, 1.77933097e-01,
2.39515111e-01, -1.09844141e-01, 3.12697500e-01,
2.76977926e-01, -5.68868890e-02, -9.18411240e-02,
9.21063572e-02]], dtype=float32):>,)>, output=Array([[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.3046113 ,
0.7694422 , 0.14696543],
[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.3046113 ,
0.7694422 , 0.14696547],
[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.30461127,
0.7694422 , 0.14696549],
...,
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625],
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625],
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625]], dtype=float32):>,)>,}, output=Array([[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.3046113 ,
0.7694422 , 0.14696543],
[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.3046113 ,
0.7694422 , 0.14696547],
[ 0.80585057, -0.7382735 , 1.258914 , ..., 0.30461127,
0.7694422 , 0.14696549],
...,
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625],
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625],
[-0.01422109, -0.9490089 , 0.17385475, ..., -0.5604846 ,
0.02295847, -0.47840625]], dtype=float32):>,)>, 'Linear_0': input='weights_or_input': 42x0e+14x1e+8x2e
[[ 0.00609185 -0.00317768 -0.00521756 ... 0.00021218 0.00682142
0.00276164]
[ 0.00609185 -0.00317768 -0.00521756 ... 0.00021218 -0.00682142
0.00276166]
[ 0.00609184 -0.00317768 -0.00521755 ... 0.00021219 -0.00682142
0.00276164]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00609185, -0.00317768, -0.00521756, ..., 0.00021218,
0.00682142, 0.00276164],
[ 0.00609185, -0.00317768, -0.00521756, ..., 0.00021218,
-0.00682142, 0.00276166],
[ 0.00609184, -0.00317768, -0.00521755, ..., 0.00021219,
-0.00682142, 0.00276164],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, params=#=2024, w[0,0] 42x0e,42x0e=Array([[-0.09137946, 2.5628996 , -1.0069333 , ..., -0.20392592,
-0.30027637, -1.5703651 ],
[ 1.2967609 , 1.3547949 , -0.3077918 , ..., -1.5977606 ,
1.1510273 , -0.69028217],
[ 0.4262043 , -0.9869622 , 0.6111858 , ..., -0.26018268,
-0.6741678 , 0.93089205],
...,
[-0.8391385 , 1.7070186 , -1.27063 , ..., 0.5932598 ,
-0.978556 , -0.12995657],
[ 0.6662011 , -0.81331885, 0.31959626, ..., 0.5336734 ,
0.8732674 , -1.8962247 ],
[ 0.965216 , 0.33714688, -1.7817831 , ..., -2.0571163 ,
-0.70515287, -1.235471 ]], dtype=float32):>, w[1,1] 14x1e,14x1e=Array([[-0.23302369, -0.5181007 , 0.891499 , 2.1908839 , -0.00584863,
-0.02704493, 0.07968911, 0.14461759, -0.07985923, -0.15228446,
-1.6886613 , 2.366294 , -0.99924874, -1.1399937 ],
[ 0.65659016, 0.90226173, 0.21762258, -0.8600763 , -1.0149279 ,
-0.46547633, 0.81110984, 0.2362861 , 1.0614411 , 1.5616652 ,
0.8749693 , -0.33338338, 1.8136635 , 1.087199 ],
[ 0.48713264, 0.30968177, -1.597842 , 0.15942553, 2.3211741 ,
1.2168036 , 1.1015903 , 0.5788648 , 0.26720098, 0.63509864,
-0.63437146, -1.1713663 , 1.8083926 , 1.4310291 ],
[ 0.5827992 , 0.31144533, 1.3100617 , 0.18009983, -0.9074706 ,
-1.3261123 , 0.7965933 , -0.9258348 , 0.8201388 , -1.8493836 ,
0.4669612 , 0.22040625, -0.50567025, -0.38086927],
[ 1.3579715 , 1.425243 , -0.76479006, -0.07505306, 0.253305 ,
-0.12046561, -1.0138636 , -2.0414999 , -0.9687636 , 0.07116902,
0.5814498 , -0.12664758, 0.29805505, 0.5359826 ],
[-0.27664796, -0.2277023 , 0.07328595, -0.61585516, -0.01692552,
0.41674653, 0.15038918, 0.39251727, -2.2173145 , 0.5428737 ,
0.69988227, 0.8866308 , 2.4927528 , -1.4090725 ],
[ 1.974789 , 1.3570299 , 0.20354167, 2.4982336 , 1.2475269 ,
0.33088526, 0.3253296 , 0.68094087, 0.19802941, 0.6527997 ,
-0.5323066 , -0.09934012, 0.48454624, -0.8345717 ],
[ 0.17872666, -0.18376228, -0.41237223, 1.1205825 , 1.0485171 ,
-0.11966894, -1.2989379 , -1.4929941 , 1.0871181 , 1.0117643 ,
0.04810008, -0.67899513, -0.35764346, 0.70563823],
[-1.5490344 , -0.12857683, -0.64421624, -0.335352 , 0.02101507,
-1.28995 , -1.7265507 , 1.0704557 , 0.32471627, -0.64989763,
2.0196285 , -0.94246656, -0.9065026 , 0.6764497 ],
[ 1.1075568 , 1.6812309 , -0.43827993, 0.90608245, -1.4182389 ,
-1.48636 , 1.8667086 , -1.6037778 , -1.6489383 , 1.8312478 ,
-1.262828 , 0.40019786, -1.0553217 , 0.81535435],
[-0.7914238 , -1.114038 , 1.3428053 , -1.7652521 , 0.7962482 ,
-0.82779783, 1.0809903 , -0.6892981 , -0.6115934 , 0.87413883,
-1.9393747 , 0.7537149 , -1.8441645 , -0.5761286 ],
[-0.03839851, -0.04638372, -0.38422823, -1.5140374 , -0.22713378,
-1.9825373 , -1.5276322 , -0.36719915, -0.33838207, -0.12497508,
0.79839414, -0.5063505 , 1.4017372 , 0.6503774 ],
[-0.49978074, -0.6047486 , -0.10998678, 0.56571794, 0.03043114,
0.1268836 , 0.6053742 , 0.36969882, -2.1814642 , 0.02026011,
0.8899261 , 0.4626365 , -1.0233332 , -0.21966554],
[ 0.36765322, -0.7244794 , -1.869059 , -0.94974583, -1.2333981 ,
0.23992056, -0.28752288, -0.40148488, 1.3103825 , -0.3170116 ,
-0.4426827 , 0.3940614 , -2.3249364 , -1.1153761 ]], dtype=float32):>, w[2,2] 8x2e,8x2e=Array([[-1.3734533 , -0.70480084, 0.8501573 , 0.82085717, -0.4550737 ,
-1.1151552 , -0.50880384, -0.7491237 ],
[-0.87768155, -0.47017616, 0.3405407 , -0.9749844 , 0.03832719,
0.316712 , -0.14743857, -1.2584321 ],
[-0.57809776, 1.2074124 , -0.6090364 , -1.0958328 , -0.4394769 ,
0.014112 , 1.7751184 , 0.02716346],
[-1.123573 , 0.24032377, 2.1357532 , -0.3344446 , -2.4958673 ,
0.7762288 , 0.3183248 , -0.52796423],
[ 0.11227739, 0.31223646, -0.99386466, 1.1569848 , 0.73100007,
-0.8996953 , 0.47191614, 0.75021744],
[-0.42602155, -0.44427806, -0.783423 , 1.0592715 , 1.2793554 ,
0.2966396 , 0.92292845, 1.5727416 ],
[ 0.1541803 , -0.59409326, -0.47240806, 0.35576066, -0.1261304 ,
-0.80945903, 0.20317438, 0.1750986 ],
[ 1.2179114 , 0.30855656, 0.02891669, -0.01709677, -1.233391 ,
0.22249964, 0.3892535 , 0.6267845 ]], dtype=float32):>,)>, output=42x0e+14x1e+8x2e
[[-0.00651554 0.01034967 0.00895069 ... -0.00032743 0.00269927
0.00130386]
[-0.00651554 0.01034966 0.00895069 ... -0.00032743 -0.00269927
0.0013039 ]
[-0.00651553 0.01034967 0.0089507 ... -0.00032739 -0.00269927
0.00130386]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00651554, 0.01034967, 0.00895069, ..., -0.00032743,
0.00269927, 0.00130386],
[-0.00651554, 0.01034966, 0.00895069, ..., -0.00032743,
-0.00269927, 0.0013039 ],
[-0.00651553, 0.01034967, 0.0089507 , ..., -0.00032739,
-0.00269927, 0.00130386],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>,}, output=42x0e+14x1e+8x2e
[[-0.00651554 0.01034967 0.00895069 ... -0.00032743 0.00269927
0.00130386]
[-0.00651554 0.01034966 0.00895069 ... -0.00032743 -0.00269927
0.0013039 ]
[-0.00651553 0.01034967 0.0089507 ... -0.00032739 -0.00269927
0.00130386]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00651554, 0.01034967, 0.00895069, ..., -0.00032743,
0.00269927, 0.00130386],
[-0.00651554, 0.01034966, 0.00895069, ..., -0.00032743,
-0.00269927, 0.0013039 ],
[-0.00651553, 0.01034967, 0.0089507 , ..., -0.00032739,
-0.00269927, 0.00130386],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'resid_ln': input='x': 42x0e+14x1e+8x2e
[[-0.00651554 0.01034967 0.00895069 ... -0.00032743 0.00269927
0.00130386]
[-0.00651554 0.01034966 0.00895069 ... -0.00032743 -0.00269927
0.0013039 ]
[-0.00651553 0.01034967 0.0089507 ... -0.00032739 -0.00269927
0.00130386]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00651554, 0.01034967, 0.00895069, ..., -0.00032743,
0.00269927, 0.00130386],
[-0.00651554, 0.01034966, 0.00895069, ..., -0.00032743,
-0.00269927, 0.0013039 ],
[-0.00651553, 0.01034967, 0.0089507 , ..., -0.00032739,
-0.00269927, 0.00130386],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, params=#=64, ln_scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32):>,)>, output=42x0e+14x1e+8x2e
[[-0.2419021 0.26448876 0.22248329 ... -0.01002653 0.08265692
0.0399267 ]
[-0.24190204 0.26448855 0.2224834 ... -0.0100264 -0.08265692
0.03992791]
[-0.24190205 0.26448867 0.22248365 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.2419021 , 0.26448876, 0.22248329, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.24190204, 0.26448855, 0.2224834 , ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.24190205, 0.26448867, 0.22248365, ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'ResidualAdapter_0': input='x': 64x0e
[[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
[ 0.02668385 0.15896109 -0.08048535 ... 0.00164835 -0.03867799
-0.09428938]
...
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]
[-0.1208031 -0.22695495 -0.07564072 ... -0.2238308 -0.3034929
-0.08798173]]key=0,): Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32), :>,}, children='64->42_0e': input='inputs': Array([[[ 0.02668385],
[ 0.15896109],
[-0.08048535],
...,
[ 0.00164835],
[-0.03867799],
[-0.09428938]],
[[ 0.02668385],
[ 0.15896109],
[-0.08048535],
...,
[ 0.00164835],
[-0.03867799],
[-0.09428938]],
[[ 0.02668385],
[ 0.15896109],
[-0.08048535],
...,
[ 0.00164835],
[-0.03867799],
[-0.09428938]],
...,
[[-0.1208031 ],
[-0.22695495],
[-0.07564072],
...,
[-0.2238308 ],
[-0.3034929 ],
[-0.08798173]],
[[-0.1208031 ],
[-0.22695495],
[-0.07564072],
...,
[-0.2238308 ],
[-0.3034929 ],
[-0.08798173]],
[[-0.1208031 ],
[-0.22695495],
[-0.07564072],
...,
[-0.2238308 ],
[-0.3034929 ],
[-0.08798173]]], dtype=float32):>,}, params=#=2688, kernel=Array([[-0.07920904, 0.09743841, -0.15760393, ..., -0.03190419,
0.18500672, -0.22597799],
[ 0.16017573, 0.01261836, -0.15161681, ..., -0.11481161,
-0.01237781, -0.02308484],
[-0.20209268, -0.08266207, -0.03450214, ..., -0.01841416,
-0.21113952, -0.02277938],
...,
[-0.00556626, 0.1265749 , 0.08346546, ..., 0.25701725,
0.06141955, 0.05101364],
[-0.11991848, -0.10978717, 0.18031135, ..., -0.11996888,
-0.04246823, -0.08142848],
[-0.09521133, -0.07060309, -0.12696336, ..., -0.22223696,
-0.11120731, 0.01468545]], dtype=float32):>,)>, output=Array([[[ 0.07139563, 0.13567518, -0.14308925, ..., -0.23124236,
0.15970823, -0.1354123 ]],
[[ 0.07139563, 0.13567518, -0.14308925, ..., -0.23124236,
0.15970823, -0.1354123 ]],
[[ 0.07139563, 0.13567518, -0.14308925, ..., -0.23124236,
0.15970823, -0.1354123 ]],
...,
[[ 0.03709371, -0.03564092, 0.02344972, ..., -0.07406045,
-0.04630058, 0.2027607 ]],
[[ 0.03709371, -0.03564092, 0.02344972, ..., -0.07406045,
-0.04630058, 0.2027607 ]],
[[ 0.03709371, -0.03564092, 0.02344972, ..., -0.07406045,
-0.04630058, 0.2027607 ]]], dtype=float32):>,)>,}, output=42x0e+14x1e+8x2e
[[ 0.07139563 0.13567518 -0.14308925 ... 0. 0.
0. ]
[ 0.07139563 0.13567518 -0.14308925 ... 0. 0.
0. ]
[ 0.07139563 0.13567518 -0.14308925 ... 0. 0.
0. ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[ 0.07139563, 0.13567518, -0.14308925, ..., 0. ,
0. , 0. ],
[ 0.07139563, 0.13567518, -0.14308925, ..., 0. ,
0. , 0. ],
[ 0.07139563, 0.13567518, -0.14308925, ..., 0. ,
0. , 0. ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>,}, output=42x0e+14x1e+8x2e
[[-0.17050648 0.40016395 0.07939404 ... -0.01002653 0.08265692
0.0399267 ]
[-0.17050642 0.4001637 0.07939415 ... -0.0100264 -0.08265692
0.03992791]
[-0.17050642 0.40016383 0.0793944 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[-0.17050648, 0.40016395, 0.07939404, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.17050642, 0.4001637 , 0.07939415, ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.17050642, 0.40016383, 0.0793944 , ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>, None,),)>, 'layer_1': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 42x0e+14x1e+8x2e
[[-0.17050648 0.40016395 0.07939404 ... -0.01002653 0.08265692
0.0399267 ]
[-0.17050642 0.4001637 0.07939415 ... -0.0100264 -0.08265692
0.03992791]
[-0.17050642 0.40016383 0.0793944 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[-0.17050648, 0.40016395, 0.07939404, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.17050642, 0.4001637 , 0.07939415, ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.17050642, 0.40016383, 0.0793944 , ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>, 'node_species': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>, 'species_embed': Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,}, children='interaction': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 42x0e+14x1e+8x2e
[[-0.17050648 0.40016395 0.07939404 ... -0.01002653 0.08265692
0.0399267 ]
[-0.17050642 0.4001637 0.07939415 ... -0.0100264 -0.08265692
0.03992791]
[-0.17050642 0.40016383 0.0793944 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[-0.17050648, 0.40016395, 0.07939404, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.17050642, 0.4001637 , 0.07939415, ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.17050642, 0.40016383, 0.0793944 , ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,}, children='linear_intro': input='weights_or_input': 42x0e+14x1e+8x2e
[[-0.17050648 0.40016395 0.07939404 ... -0.01002653 0.08265692
0.0399267 ]
[-0.17050642 0.4001637 0.07939415 ... -0.0100264 -0.08265692
0.03992791]
[-0.17050642 0.40016383 0.0793944 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[-0.17050648, 0.40016395, 0.07939404, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.17050642, 0.4001637 , 0.07939415, ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.17050642, 0.40016383, 0.0793944 , ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, params=#=2024, w[0,0] 42x0e,42x0e=Array([[-0.9437564 , -0.7373231 , -0.10565937, ..., -0.15347199,
1.4206418 , -0.12635088],
[ 0.33079657, -0.2543346 , -0.3143749 , ..., 0.86338323,
-0.5363133 , -0.04485489],
[ 0.44902924, -1.0872252 , 0.89866537, ..., -0.5612635 ,
-0.31895906, -0.09730855],
...,
[-0.17909968, 0.46708718, 1.0508635 , ..., -0.7353951 ,
0.78218263, -0.12056134],
[-0.0035086 , -0.22287288, -0.9191797 , ..., -0.6997633 ,
-1.3232309 , -0.5470451 ],
[-0.50751835, -1.0029007 , 0.3201526 , ..., -0.5031272 ,
-1.0706912 , 1.0018215 ]], dtype=float32):>, w[1,1] 14x1e,14x1e=Array([[ 1.1614727 , 1.2480273 , -0.7704654 , -1.3391349 , 1.3500768 ,
0.32858363, -1.7546644 , 0.66505176, 0.7502384 , -0.05866462,
-0.7730198 , -1.4357064 , -1.7516607 , -1.7075233 ],
[ 0.6452225 , -2.090632 , -0.91414505, 0.40276638, 1.6493465 ,
-0.83729535, 1.1459658 , -0.84389776, 0.46051082, -1.2228075 ,
0.5595544 , -0.3101086 , -0.20052868, -0.30287626],
[-0.71218437, 0.3958944 , 0.4283971 , -0.59427935, -0.93258184,
0.8413768 , 0.30736685, 0.88966864, 1.2035637 , -0.5876066 ,
0.15780391, 1.3465898 , -0.53295535, -1.2051247 ],
[-1.0457028 , 0.25284216, -0.21628673, -0.4186727 , -0.2454435 ,
-0.552498 , -1.14551 , -1.3570071 , 1.0237311 , 0.49092907,
0.17983653, -0.11645138, -0.9523089 , -0.55805224],
[ 1.1690443 , -0.38119894, 0.14042379, -0.45765063, -0.41954148,
0.37197143, 0.6460256 , 1.2588699 , -0.7026091 , 0.28135607,
-0.13176547, 0.09121997, -1.0278255 , -0.2146224 ],
[ 0.12697576, -0.8982984 , 1.1811503 , 0.89561605, -0.46632802,
-0.05594215, 0.80818695, 0.12257929, 0.43226945, -1.2712164 ,
-1.2479887 , -0.21832499, -1.4196813 , 0.45544842],
[ 2.0398638 , -1.6008942 , 0.22649929, 2.0554864 , -0.29308254,
0.34688696, -0.6686714 , -1.9671822 , 0.8928897 , -0.51799136,
-0.4564329 , 2.5349483 , 1.3789272 , 0.49224874],
[ 0.44596225, 1.388066 , 2.0389674 , 0.14383043, 0.44201824,
1.4223479 , 0.44242486, -1.7352164 , -0.98409736, 2.0425777 ,
0.29267097, 1.2651945 , -0.50300103, 0.7975486 ],
[ 1.105337 , 0.0975561 , 0.33318546, 0.34341067, -1.9508867 ,
-1.1155204 , -0.672583 , 0.603176 , -1.9819918 , -0.3344987 ,
0.05306494, -0.06778801, -0.6059689 , -0.6624143 ],
[ 0.17413926, 2.3397052 , 1.389284 , 1.8064569 , 0.01592302,
-0.7124447 , 0.65297794, 0.58392537, 2.3885803 , 0.7977761 ,
0.01399874, -1.9638354 , -0.5988473 , -0.08720936],
[-1.325825 , 2.3861477 , -0.08173112, 0.2318904 , -0.02630361,
-1.1673496 , 0.46071982, 1.4307913 , -0.14945918, -1.3821374 ,
0.20926018, -1.1580827 , -1.8123952 , 1.197906 ],
[ 1.1973484 , -1.0803235 , 1.190191 , -0.12835808, 0.09834486,
1.2073275 , 0.28126156, 0.5606919 , 2.2251313 , -1.4514166 ,
-0.05231043, -0.4304393 , 0.0637676 , 0.17302687],
[ 0.22034684, -0.8664613 , -0.6505869 , -0.8603708 , -0.9916646 ,
-0.7520616 , -1.077296 , -1.1044147 , 0.04941732, 1.0408416 ,
0.60972416, -1.7191799 , -0.5259832 , -0.04641453],
[-0.5188639 , -1.1565375 , 0.49527663, -1.3225787 , -0.5677822 ,
0.5365325 , 0.3540654 , 0.42468256, 0.8883537 , 0.3845569 ,
-0.9567712 , 0.9872624 , 1.430364 , 0.13487782]], dtype=float32):>, w[2,2] 8x2e,8x2e=Array([[-0.43363973, -2.1674972 , 0.37483665, 0.2940152 , -1.6449925 ,
-0.95496446, 0.10382193, -0.49600998],
[ 0.3451145 , -1.65782 , -0.23688918, -1.1928581 , 0.88736624,
-0.5579984 , 0.9177612 , 0.20249815],
[ 0.39709368, -1.1624568 , -1.6325058 , -0.8220629 , 1.4532245 ,
-1.6092293 , 1.0219669 , 0.9645977 ],
[-0.55189663, -0.5044224 , -0.49967137, 0.17532194, -0.53281343,
0.24574915, -0.47733054, 1.110371 ],
[ 0.2061791 , 0.943759 , -0.6234479 , -0.26159492, -0.8215837 ,
0.49227005, 0.6398137 , -1.322691 ],
[ 0.3203498 , -0.45629564, -0.44637567, -0.52311915, -1.5396084 ,
-0.97170013, -0.54127765, 2.6289122 ],
[ 1.691275 , 0.36002734, -0.68464273, 1.1465433 , -1.2550778 ,
-1.6420689 , 0.5918922 , 0.44062465],
[ 1.4189862 , 1.0839026 , 0.0432342 , -0.35113442, -0.21468508,
-0.48242176, -0.11105376, 1.5714545 ]], dtype=float32):>,)>, output=42x0e+14x1e+8x2e
[[ 0.39625508 0.16749014 0.33186224 ... 0.00546897 0.11267531
0.06266429]
[ 0.39625502 0.16749008 0.33186233 ... 0.00546915 -0.11267531
0.06266439]
[ 0.39625493 0.16748987 0.33186212 ... 0.00546996 -0.11267531
0.06266429]
...
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]]key=0,): Array([[ 0.39625508, 0.16749014, 0.33186224, ..., 0.00546897,
0.11267531, 0.06266429],
[ 0.39625502, 0.16749008, 0.33186233, ..., 0.00546915,
-0.11267531, 0.06266439],
[ 0.39625493, 0.16748987, 0.33186212, ..., 0.00546996,
-0.11267531, 0.06266429],
...,
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ],
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ],
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'SevenNetConv_0': input='vectors': 1x0e+1x1e+1x2e
[[[ 1. -0.9357057 -0.45666325 ... -0.8848778 0.8160343
0.671555 ]
[ 1. 0.07513384 1.730331 ... 2.2294104 -0.03929481
-0.00344416]
[ 1. -0.829883 -0.6267714 ... -0.6788228 -1.1207505
0.7937976 ]
...
[ 1. 1.0278528 1.3939806 ... 1.0545096 -0.03300212
-0.6817387 ]
[ 1. 0.00728687 -0.9819291 ... -0.04004242 -1.8087059
1.3140441 ]
[ 1. 1.4244379 -0.8501073 ... -0.31005016 -0.546866
-1.1494559 ]]
[[ 1. 0.93570566 -0.45666313 ... -0.88487786 -0.8160341
0.6715551 ]
[ 1. -0.0751342 1.7303311 ... 2.2294104 0.03929502
-0.00344419]
[ 1. 0.82988256 -0.6267716 ... -0.67882234 1.120751
0.793798 ]
...
[ 1. -1.0278528 1.3939805 ... 1.0545094 0.03300202
-0.6817387 ]
[ 1. -0.00728687 -0.98192924 ... -0.04004198 1.8087058
1.3140436 ]
[ 1. -1.4244381 -0.8501072 ... -0.31005067 0.5468661
-1.1494563 ]]
[[ 1. 0.93570596 0.4566635 ... -0.88487744 -0.81603456
0.67155415]
[ 1. -0.07513385 -1.7303311 ... 2.2294104 0.03929329
-0.00344417]
[ 1. 0.82988274 0.6267717 ... -0.6788224 1.1207511
0.79379785]
...
[ 1. -1.0278533 -1.3939804 ... 1.0545088 0.03300175
-0.6817393 ]
[ 1. -0.00728687 0.981929 ... -0.04004287 1.8087059
1.3140446 ]
[ 1. -1.4244381 0.8501073 ... -0.31005055 0.5468663
-1.1494561 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , -0.9357057 , -0.45666325, ..., -0.8848778 ,
0.8160343 , 0.671555 ],
[ 1. , 0.07513384, 1.730331 , ..., 2.2294104 ,
-0.03929481, -0.00344416],
[ 1. , -0.829883 , -0.6267714 , ..., -0.6788228 ,
-1.1207505 , 0.7937976 ],
...,
[ 1. , 1.0278528 , 1.3939806 , ..., 1.0545096 ,
-0.03300212, -0.6817387 ],
[ 1. , 0.00728687, -0.9819291 , ..., -0.04004242,
-1.8087059 , 1.3140441 ],
[ 1. , 1.4244379 , -0.8501073 , ..., -0.31005016,
-0.546866 , -1.1494559 ]],
[[ 1. , 0.93570566, -0.45666313, ..., -0.88487786,
-0.8160341 , 0.6715551 ],
[ 1. , -0.0751342 , 1.7303311 , ..., 2.2294104 ,
0.03929502, -0.00344419],
[ 1. , 0.82988256, -0.6267716 , ..., -0.67882234,
1.120751 , 0.793798 ],
...,
[ 1. , -1.0278528 , 1.3939805 , ..., 1.0545094 ,
0.03300202, -0.6817387 ],
[ 1. , -0.00728687, -0.98192924, ..., -0.04004198,
1.8087058 , 1.3140436 ],
[ 1. , -1.4244381 , -0.8501072 , ..., -0.31005067,
0.5468661 , -1.1494563 ]],
[[ 1. , 0.93570596, 0.4566635 , ..., -0.88487744,
-0.81603456, 0.67155415],
[ 1. , -0.07513385, -1.7303311 , ..., 2.2294104 ,
0.03929329, -0.00344417],
[ 1. , 0.82988274, 0.6267717 , ..., -0.6788224 ,
1.1207511 , 0.79379785],
...,
[ 1. , -1.0278533 , -1.3939804 , ..., 1.0545088 ,
0.03300175, -0.6817393 ],
[ 1. , -0.00728687, 0.981929 , ..., -0.04004287,
1.8087059 , 1.3140446 ],
[ 1. , -1.4244381 , 0.8501073 , ..., -0.31005055,
0.5468663 , -1.1494561 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :>, 'node_feats': 42x0e+14x1e+8x2e
[[ 0.39625508 0.16749014 0.33186224 ... 0.00546897 0.11267531
0.06266429]
[ 0.39625502 0.16749008 0.33186233 ... 0.00546915 -0.11267531
0.06266439]
[ 0.39625493 0.16748987 0.33186212 ... 0.00546996 -0.11267531
0.06266429]
...
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]
[-0.16782767 0.04393858 -0.09223274 ... 0. 0.
0. ]]key=0,): Array([[ 0.39625508, 0.16749014, 0.33186224, ..., 0.00546897,
0.11267531, 0.06266429],
[ 0.39625502, 0.16749008, 0.33186233, ..., 0.00546915,
-0.11267531, 0.06266439],
[ 0.39625493, 0.16748987, 0.33186212, ..., 0.00546996,
-0.11267531, 0.06266429],
...,
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ],
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ],
[-0.16782767, 0.04393858, -0.09223274, ..., 0. ,
0. , 0. ]], dtype=float32), :>, 'radial_embedding': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>, 'receivers': Array([[216, 153, 82, ..., 178, 99, 123],
[217, 155, 80, ..., 176, 98, 122],
[218, 154, 81, ..., 177, 97, 121],
...,
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388],
[388, 388, 388, ..., 388, 388, 388]], dtype=uint32):>,}, children='LazyInMLP_0': input='x': 16x0e
[[[2.08870246e-20 1.90958130e-13 5.35457652e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16946779e-20 1.19752041e-13 3.76099614e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07113540e-20 1.11575191e-13 3.56463836e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31235951e-32 ... 4.15392078e-16
1.09356498e-23 8.83007846e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08868646e-20 1.90956680e-13 5.35454596e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07118451e-20 1.11579244e-13 3.56474033e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31232953e-32 ... 4.15398430e-16
1.09358580e-23 8.83028050e-33]
[0.00000000e+00 0.00000000e+00 6.52239155e-36 ... 5.95891134e-14
4.91981098e-21 1.24584601e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
[[2.08877403e-20 1.90963971e-13 5.35469908e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.16949016e-20 1.19753870e-13 3.76103948e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.07115592e-20 1.11576899e-13 3.56467922e-08 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[0.00000000e+00 3.46120721e-43 1.31238963e-32 ... 4.15385778e-16
1.09354407e-23 8.82987643e-33]
[0.00000000e+00 0.00000000e+00 6.52219209e-36 ... 5.95900214e-14
4.91990509e-21 1.24587453e-29]
[0.00000000e+00 0.00000000e+00 4.21187495e-38 ... 1.15994464e-12
1.98385376e-19 1.04067836e-27]]
...
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
[[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
...
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[1.00000000e+00 1.75130755e-01 9.40696802e-04 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]]key=0,): Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32), :>,}, children='Dense_0': input='inputs': Array([[[2.08870246e-20, 1.90958130e-13, 5.35457652e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16946779e-20, 1.19752041e-13, 3.76099614e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07113540e-20, 1.11575191e-13, 3.56463836e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31235951e-32, ...,
4.15392078e-16, 1.09356498e-23, 8.83007846e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08868646e-20, 1.90956680e-13, 5.35454596e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07118451e-20, 1.11579244e-13, 3.56474033e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31232953e-32, ...,
4.15398430e-16, 1.09358580e-23, 8.83028050e-33],
[0.00000000e+00, 0.00000000e+00, 6.52239155e-36, ...,
5.95891134e-14, 4.91981098e-21, 1.24584601e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
[[2.08877403e-20, 1.90963971e-13, 5.35469908e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.16949016e-20, 1.19753870e-13, 3.76103948e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.07115592e-20, 1.11576899e-13, 3.56467922e-08, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[0.00000000e+00, 3.46120721e-43, 1.31238963e-32, ...,
4.15385778e-16, 1.09354407e-23, 8.82987643e-33],
[0.00000000e+00, 0.00000000e+00, 6.52219209e-36, ...,
5.95900214e-14, 4.91990509e-21, 1.24587453e-29],
[0.00000000e+00, 0.00000000e+00, 4.21187495e-38, ...,
1.15994464e-12, 1.98385376e-19, 1.04067836e-27]],
...,
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]],
[[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
...,
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[1.00000000e+00, 1.75130755e-01, 9.40696802e-04, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]], dtype=float32):>,}, params=#=1024, kernel=Array([[-0.11241984, 0.03942039, 0.27701008, ..., -0.0917427 ,
-0.02097219, 0.13423194],
[-0.23743476, 0.29689342, 0.22044863, ..., -0.09596665,
-0.2307145 , 0.21455377],
[-0.18931031, -0.05400281, -0.15956768, ..., 0.13589323,
-0.10310976, 0.143225 ],
...,
[-0.2604794 , 0.08742005, -0.13457353, ..., 0.2916095 ,
0.0275303 , 0.05086919],
[ 0.24729219, 0.00043886, 0.04033604, ..., -0.10380144,
0.33051333, -0.20450684],
[ 0.1460626 , 0.03374685, -0.02660245, ..., 0.27881008,
0.07333156, -0.1025686 ]], dtype=float32):>,)>, output=Array([[[-0.07286622, -0.14955674, -0.06197986, ..., -0.14663026,
0.09395145, 0.13588038],
[-0.07322823, -0.15518385, -0.06246 , ..., -0.1407786 ,
0.08666754, 0.14177121],
[-0.07326075, -0.15607512, -0.06250108, ..., -0.13986033,
0.08556924, 0.14264955],
...,
[-0.32611212, -0.12235753, 0.08494989, ..., -0.00066819,
-0.1297158 , 0.05001796],
[-0.30846345, 0.02036613, 0.05190555, ..., 0.14539334,
-0.08596624, -0.02082835],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
[[-0.07286623, -0.14955682, -0.06197987, ..., -0.14663017,
0.09395134, 0.13588047],
[-0.07322822, -0.15518367, -0.06245999, ..., -0.1407788 ,
0.08666777, 0.14177105],
[-0.07326073, -0.15607464, -0.06250106, ..., -0.13986082,
0.08556981, 0.14264908],
...,
[-0.32611203, -0.12235704, 0.08494978, ..., -0.0006677 ,
-0.12971564, 0.05001766],
[-0.30846348, 0.02036577, 0.05190564, ..., 0.14539295,
-0.08596635, -0.02082823],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
[[-0.07286619, -0.1495564 , -0.06197982, ..., -0.1466306 ,
0.0939519 , 0.13588001],
[-0.07322822, -0.15518367, -0.06245999, ..., -0.1407788 ,
0.08666777, 0.14177105],
[-0.07326074, -0.15607494, -0.06250107, ..., -0.13986053,
0.08556947, 0.14264937],
...,
[-0.32611218, -0.12235802, 0.08495 , ..., -0.00066867,
-0.12971593, 0.05001824],
[-0.30846345, 0.02036613, 0.05190555, ..., 0.14539334,
-0.08596624, -0.02082835],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
...,
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]],
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]],
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]]], dtype=float32):>,)>,}, output=64x0e
[[[-0.07286622 -0.14955674 -0.06197986 ... -0.14663026 0.09395145
0.13588038]
[-0.07322823 -0.15518385 -0.06246 ... -0.1407786 0.08666754
0.14177121]
[-0.07326075 -0.15607512 -0.06250108 ... -0.13986033 0.08556924
0.14264955]
...
[-0.32611212 -0.12235753 0.08494989 ... -0.00066819 -0.1297158
0.05001796]
[-0.30846345 0.02036613 0.05190555 ... 0.14539334 -0.08596624
-0.02082835]
[-0.29443455 0.07504825 0.03768308 ... 0.20917223 -0.06939367
-0.02209714]]
[[-0.07286623 -0.14955682 -0.06197987 ... -0.14663017 0.09395134
0.13588047]
[-0.07322822 -0.15518367 -0.06245999 ... -0.1407788 0.08666777
0.14177105]
[-0.07326073 -0.15607464 -0.06250106 ... -0.13986082 0.08556981
0.14264908]
...
[-0.32611203 -0.12235704 0.08494978 ... -0.0006677 -0.12971564
0.05001766]
[-0.30846348 0.02036577 0.05190564 ... 0.14539295 -0.08596635
-0.02082823]
[-0.29443455 0.07504825 0.03768308 ... 0.20917223 -0.06939367
-0.02209714]]
[[-0.07286619 -0.1495564 -0.06197982 ... -0.1466306 0.0939519
0.13588001]
[-0.07322822 -0.15518367 -0.06245999 ... -0.1407788 0.08666777
0.14177105]
[-0.07326074 -0.15607494 -0.06250107 ... -0.13986053 0.08556947
0.14264937]
...
[-0.32611218 -0.12235802 0.08495 ... -0.00066867 -0.12971593
0.05001824]
[-0.30846345 0.02036613 0.05190555 ... 0.14539334 -0.08596624
-0.02082835]
[-0.29443455 0.07504825 0.03768308 ... 0.20917223 -0.06939367
-0.02209714]]
...
[[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
...
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]]
[[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
...
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]]
[[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
...
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]
[-0.15418003 0.0913647 0.3154673 ... -0.10842158 -0.06147438
0.17194167]]]key=0,): Array([[[-0.07286622, -0.14955674, -0.06197986, ..., -0.14663026,
0.09395145, 0.13588038],
[-0.07322823, -0.15518385, -0.06246 , ..., -0.1407786 ,
0.08666754, 0.14177121],
[-0.07326075, -0.15607512, -0.06250108, ..., -0.13986033,
0.08556924, 0.14264955],
...,
[-0.32611212, -0.12235753, 0.08494989, ..., -0.00066819,
-0.1297158 , 0.05001796],
[-0.30846345, 0.02036613, 0.05190555, ..., 0.14539334,
-0.08596624, -0.02082835],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
[[-0.07286623, -0.14955682, -0.06197987, ..., -0.14663017,
0.09395134, 0.13588047],
[-0.07322822, -0.15518367, -0.06245999, ..., -0.1407788 ,
0.08666777, 0.14177105],
[-0.07326073, -0.15607464, -0.06250106, ..., -0.13986082,
0.08556981, 0.14264908],
...,
[-0.32611203, -0.12235704, 0.08494978, ..., -0.0006677 ,
-0.12971564, 0.05001766],
[-0.30846348, 0.02036577, 0.05190564, ..., 0.14539295,
-0.08596635, -0.02082823],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
[[-0.07286619, -0.1495564 , -0.06197982, ..., -0.1466306 ,
0.0939519 , 0.13588001],
[-0.07322822, -0.15518367, -0.06245999, ..., -0.1407788 ,
0.08666777, 0.14177105],
[-0.07326074, -0.15607494, -0.06250107, ..., -0.13986053,
0.08556947, 0.14264937],
...,
[-0.32611218, -0.12235802, 0.08495 , ..., -0.00066867,
-0.12971593, 0.05001824],
[-0.30846345, 0.02036613, 0.05190555, ..., 0.14539334,
-0.08596624, -0.02082835],
[-0.29443455, 0.07504825, 0.03768308, ..., 0.20917223,
-0.06939367, -0.02209714]],
...,
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]],
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]],
[[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
...,
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167],
[-0.15418003, 0.0913647 , 0.3154673 , ..., -0.10842158,
-0.06147438, 0.17194167]]], dtype=float32), :>,)>,}, output=64x0e
[[-0.11364639 -0.02628255 0.01588504 ... 0.00404463 -0.00072424
-0.00101662]
[-0.1136464 -0.02628255 0.01588504 ... 0.00404469 -0.00072424
-0.0010166 ]
[-0.11364637 -0.02628257 0.01588506 ... 0.00404459 -0.00072423
-0.00101666]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.11364639, -0.02628255, 0.01588504, ..., 0.00404463,
-0.00072424, -0.00101662],
[-0.1136464 , -0.02628255, 0.01588504, ..., 0.00404469,
-0.00072424, -0.0010166 ],
[-0.11364637, -0.02628257, 0.01588506, ..., 0.00404459,
-0.00072423, -0.00101666],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'linear_outro': input='weights_or_input': 64x0e
[[-0.11364639 -0.02628255 0.01588504 ... 0.00404463 -0.00072424
-0.00101662]
[-0.1136464 -0.02628255 0.01588504 ... 0.00404469 -0.00072424
-0.0010166 ]
[-0.11364637 -0.02628257 0.01588506 ... 0.00404459 -0.00072423
-0.00101666]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.11364639, -0.02628255, 0.01588504, ..., 0.00404463,
-0.00072424, -0.00101662],
[-0.1136464 , -0.02628255, 0.01588504, ..., 0.00404469,
-0.00072424, -0.0010166 ],
[-0.11364637, -0.02628257, 0.01588506, ..., 0.00404459,
-0.00072423, -0.00101666],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, params=#=4096, w[0,0] 64x0e,64x0e=Array([[ 0.57535326, 2.5397 , -0.13135676, ..., -0.6070434 ,
0.9561322 , 1.1101446 ],
[-0.24770263, -1.8274362 , 1.6751906 , ..., -0.7568171 ,
-1.3092217 , 0.92734873],
[ 0.03861037, 0.5306088 , 0.18995585, ..., -0.46038175,
0.9535147 , -1.1047027 ],
...,
[ 0.3595654 , -1.1171188 , 1.9702958 , ..., 1.0499529 ,
-1.3601849 , 0.9640759 ],
[ 0.55107754, 0.97608244, 0.43133652, ..., -1.8373475 ,
-0.21804775, -0.1775932 ],
[ 0.11132542, 0.5663938 , -1.7796549 , ..., 0.52715075,
0.76737607, 0.05876654]], dtype=float32):>,)>, output=64x0e
[[ 0.00083467 -0.03402488 -0.01186005 ... 0.00454617 -0.01975407
-0.01692413]
[ 0.00083468 -0.03402489 -0.01186004 ... 0.00454618 -0.01975408
-0.01692412]
[ 0.00083467 -0.03402486 -0.01186004 ... 0.00454616 -0.01975407
-0.01692414]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00083467, -0.03402488, -0.01186005, ..., 0.00454617,
-0.01975407, -0.01692413],
[ 0.00083468, -0.03402489, -0.01186004, ..., 0.00454618,
-0.01975408, -0.01692412],
[ 0.00083467, -0.03402486, -0.01186004, ..., 0.00454616,
-0.01975407, -0.01692414],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>,}, output=64x0e
[[ 0.00083467 -0.03402488 -0.01186005 ... 0.00454617 -0.01975407
-0.01692413]
[ 0.00083468 -0.03402489 -0.01186004 ... 0.00454618 -0.01975408
-0.01692412]
[ 0.00083467 -0.03402486 -0.01186004 ... 0.00454616 -0.01975407
-0.01692414]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00083467, -0.03402488, -0.01186005, ..., 0.00454617,
-0.01975407, -0.01692413],
[ 0.00083468, -0.03402489, -0.01186004, ..., 0.00454618,
-0.01975408, -0.01692412],
[ 0.00083467, -0.03402486, -0.01186004, ..., 0.00454616,
-0.01975407, -0.01692414],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'self_connection': input='node_feats': 64x0e
[[ 0.00083467 -0.03402488 -0.01186005 ... 0.00454617 -0.01975407
-0.01692413]
[ 0.00083468 -0.03402489 -0.01186004 ... 0.00454618 -0.01975408
-0.01692412]
[ 0.00083467 -0.03402486 -0.01186004 ... 0.00454616 -0.01975407
-0.01692414]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00083467, -0.03402488, -0.01186005, ..., 0.00454617,
-0.01975407, -0.01692413],
[ 0.00083468, -0.03402489, -0.01186004, ..., 0.00454618,
-0.01975408, -0.01692412],
[ 0.00083467, -0.03402486, -0.01186004, ..., 0.00454616,
-0.01975407, -0.01692414],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>, 'node_specie': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>, 'species_embed': Array([[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
[ 0.02668385, 0.15896109, -0.08048535, ..., 0.00164835,
-0.03867799, -0.09428938],
...,
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173],
[-0.1208031 , -0.22695495, -0.07564072, ..., -0.2238308 ,
-0.3034929 , -0.08798173]], dtype=float32):>,}, children='self_mlp': input='x': Array([[ 0.00083467, -0.03402488, -0.01186005, ..., 0.00454617,
-0.01975407, -0.01692413],
[ 0.00083468, -0.03402489, -0.01186004, ..., 0.00454618,
-0.01975408, -0.01692412],
[ 0.00083467, -0.03402486, -0.01186004, ..., 0.00454616,
-0.01975407, -0.01692414],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, children='Dense_0': input='inputs': Array([[ 0.00083467, -0.03402488, -0.01186005, ..., 0.00454617,
-0.01975407, -0.01692413],
[ 0.00083468, -0.03402489, -0.01186004, ..., 0.00454618,
-0.01975408, -0.01692412],
[ 0.00083467, -0.03402486, -0.01186004, ..., 0.00454616,
-0.01975407, -0.01692414],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, params=#=4096, kernel=Array([[-0.06958146, -0.11491764, 0.05689784, ..., 0.21155909,
0.0240686 , -0.07760036],
[-0.13731778, -0.17114809, -0.06211331, ..., 0.1090192 ,
0.14521338, 0.01075325],
[-0.0990229 , -0.2329272 , -0.0417339 , ..., -0.00056148,
0.05935917, -0.15156996],
...,
[-0.1274116 , 0.11570005, 0.17948832, ..., -0.06654623,
-0.1171699 , -0.2639797 ],
[ 0.08853836, -0.06891993, 0.1666888 , ..., 0.05627523,
-0.15430509, 0.02994665],
[ 0.17138524, 0.00735729, -0.15183677, ..., 0.14177264,
-0.07123093, 0.12759744]], dtype=float32):>,)>, output=Array([[-0.03049684, -0.00291995, 0.02398385, ..., -0.01416009,
0.01034274, 0.01176827],
[-0.03049687, -0.00291996, 0.02398386, ..., -0.01416008,
0.01034274, 0.01176825],
[-0.03049688, -0.00291993, 0.02398384, ..., -0.01416005,
0.01034276, 0.01176826],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,)>, 'Dropout_0': input='inputs': Array([[-0.01501592, -0.00145784, 0.01213573, ..., -0.00702992,
0.00519811, 0.00591876],
[-0.01501594, -0.00145785, 0.01213573, ..., -0.00702991,
0.00519811, 0.00591875],
[-0.01501594, -0.00145784, 0.01213572, ..., -0.0070299 ,
0.00519812, 0.00591875],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, output=Array([[-0.01501592, -0.00145784, 0.01213573, ..., -0.00702992,
0.00519811, 0.00591876],
[-0.01501594, -0.00145785, 0.01213573, ..., -0.00702991,
0.00519811, 0.00591875],
[-0.01501594, -0.00145784, 0.01213572, ..., -0.0070299 ,
0.00519812, 0.00591875],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,)>, 'LayerNorm_0': input='x': Array([[-0.01501592, -0.00145784, 0.01213573, ..., -0.00702992,
0.00519811, 0.00591876],
[-0.01501594, -0.00145785, 0.01213573, ..., -0.00702991,
0.00519811, 0.00591875],
[-0.01501594, -0.00145784, 0.01213572, ..., -0.0070299 ,
0.00519812, 0.00591875],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, params=#=64, scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32):>,)>, output=Array([[-1.5890433 , -0.09060057, 1.4117649 , ..., -0.7064278 ,
0.64501774, 0.724664 ],
[-1.5890443 , -0.09060109, 1.4117649 , ..., -0.706427 ,
0.6450175 , 0.7246626 ],
[-1.5890459 , -0.09059965, 1.4117643 , ..., -0.70642585,
0.6450187 , 0.7246631 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,)>, 'Dense_1': input='inputs': Array([[-1.5890433 , -0.09060057, 1.4117649 , ..., -0.7064278 ,
0.64501774, 0.724664 ],
[-1.5890443 , -0.09060109, 1.4117649 , ..., -0.706427 ,
0.6450175 , 0.7246626 ],
[-1.5890459 , -0.09059965, 1.4117643 , ..., -0.70642585,
0.6450187 , 0.7246631 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, params=#=4096, kernel=Array([[-0.2377826 , 0.21778454, -0.12765573, ..., 0.03906341,
0.07669549, -0.16993321],
[ 0.14214842, 0.09377062, -0.0825526 , ..., -0.10012055,
0.00217365, 0.01037766],
[-0.08557843, 0.00633491, 0.04984623, ..., 0.2730145 ,
-0.07432125, -0.03510907],
...,
[-0.01751295, -0.01158033, 0.12748338, ..., -0.05337953,
0.00169546, -0.05808586],
[ 0.1502636 , 0.0992592 , 0.11271845, ..., -0.05087252,
0.17387667, 0.03925362],
[ 0.02444309, -0.12427627, -0.1417498 , ..., -0.13208574,
-0.0969074 , -0.27623922]], dtype=float32):>,)>, output=Array([[ 0.42569885, -0.28983888, -1.2538655 , ..., -0.68762624,
-2.8243294 , 2.1758285 ],
[ 0.42569962, -0.289839 , -1.253865 , ..., -0.6876264 ,
-2.8243296 , 2.1758294 ],
[ 0.42569935, -0.28984046, -1.2538648 , ..., -0.68762636,
-2.8243284 , 2.1758294 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,)>,}, output=Array([[ 0.42569885, -0.28983888, -1.2538655 , ..., -0.68762624,
-2.8243294 , 2.1758285 ],
[ 0.42569962, -0.289839 , -1.253865 , ..., -0.6876264 ,
-2.8243296 , 2.1758294 ],
[ 0.42569935, -0.28984046, -1.2538648 , ..., -0.68762636,
-2.8243284 , 2.1758294 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,)>, 'Linear_0': input='weights_or_input': Array([[ 0.42569885, -0.28983888, -1.2538655 , ..., -0.68762624,
-2.8243294 , 2.1758285 ],
[ 0.42569962, -0.289839 , -1.253865 , ..., -0.6876264 ,
-2.8243296 , 2.1758294 ],
[ 0.42569935, -0.28984046, -1.2538648 , ..., -0.68762636,
-2.8243284 , 2.1758294 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32):>,}, params=#=4096, w[0,0] 64x0e,64x0e=Array([[-1.9590285 , -0.5894596 , -0.3906604 , ..., -0.9301631 ,
-0.6084556 , 0.23754641],
[-0.8134961 , 1.4233882 , -0.02487153, ..., -1.0719723 ,
-0.5839326 , -0.9470744 ],
[ 0.4334623 , 0.5664843 , 1.2671087 , ..., -0.40893623,
0.25260338, -0.74831384],
...,
[ 0.65542144, -0.37395188, 0.37244862, ..., -0.55726016,
-0.07185011, 1.3412169 ],
[-0.7136356 , -0.28322697, 0.31961825, ..., -1.396129 ,
0.5271611 , 0.3448595 ],
[-0.5843374 , 0.40660977, 1.0144186 , ..., 1.2697339 ,
0.26319528, 0.34148955]], dtype=float32):>,)>, output=64x0e
[[ 1.4536716 0.39682978 -0.68612206 ... 2.9477282 0.24927925
0.6303859 ]
[ 1.4536715 0.39682907 -0.68612105 ... 2.947728 0.24927832
0.63038677]
[ 1.453671 0.39682758 -0.686122 ... 2.9477274 0.24927641
0.63038707]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 1.4536716 , 0.39682978, -0.68612206, ..., 2.9477282 ,
0.24927925, 0.6303859 ],
[ 1.4536715 , 0.39682907, -0.68612105, ..., 2.947728 ,
0.24927832, 0.63038677],
[ 1.453671 , 0.39682758, -0.686122 , ..., 2.9477274 ,
0.24927641, 0.63038707],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>,}, output=64x0e
[[ 1.4536716 0.39682978 -0.68612206 ... 2.9477282 0.24927925
0.6303859 ]
[ 1.4536715 0.39682907 -0.68612105 ... 2.947728 0.24927832
0.63038677]
[ 1.453671 0.39682758 -0.686122 ... 2.9477274 0.24927641
0.63038707]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 1.4536716 , 0.39682978, -0.68612206, ..., 2.9477282 ,
0.24927925, 0.6303859 ],
[ 1.4536715 , 0.39682907, -0.68612105, ..., 2.947728 ,
0.24927832, 0.63038677],
[ 1.453671 , 0.39682758, -0.686122 , ..., 2.9477274 ,
0.24927641, 0.63038707],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'resid_ln': input='x': 64x0e
[[ 1.4536716 0.39682978 -0.68612206 ... 2.9477282 0.24927925
0.6303859 ]
[ 1.4536715 0.39682907 -0.68612105 ... 2.947728 0.24927832
0.63038677]
[ 1.453671 0.39682758 -0.686122 ... 2.9477274 0.24927641
0.63038707]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 1.4536716 , 0.39682978, -0.68612206, ..., 2.9477282 ,
0.24927925, 0.6303859 ],
[ 1.4536715 , 0.39682907, -0.68612105, ..., 2.947728 ,
0.24927832, 0.63038677],
[ 1.453671 , 0.39682758, -0.686122 , ..., 2.9477274 ,
0.24927641, 0.63038707],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, params=#=64, ln_scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32):>,)>, output=64x0e
[[ 1.2999743 0.38151747 -0.55963063 ... 2.5983963 0.25328746
0.58449125]
[ 1.2999742 0.38151675 -0.5596298 ... 2.598396 0.25328657
0.58449197]
[ 1.2999741 0.38151568 -0.55963063 ... 2.5983963 0.2532851
0.58449244]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 1.2999743 , 0.38151747, -0.55963063, ..., 2.5983963 ,
0.25328746, 0.58449125],
[ 1.2999742 , 0.38151675, -0.5596298 , ..., 2.598396 ,
0.25328657, 0.58449197],
[ 1.2999741 , 0.38151568, -0.55963063, ..., 2.5983963 ,
0.2532851 , 0.58449244],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :>,)>, 'ResidualAdapter_0': input='x': 42x0e+14x1e+8x2e
[[-0.17050648 0.40016395 0.07939404 ... -0.01002653 0.08265692
0.0399267 ]
[-0.17050642 0.4001637 0.07939415 ... -0.0100264 -0.08265692
0.03992791]
[-0.17050642 0.40016383 0.0793944 ... -0.01002544 -0.08265692
0.0399267 ]
...
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]
[ 0.03709371 -0.03564092 0.02344972 ... 0. 0.
0. ]]key=0,): Array([[-0.17050648, 0.40016395, 0.07939404, ..., -0.01002653,
0.08265692, 0.0399267 ],
[-0.17050642, 0.4001637 , 0.07939415, ..., -0.0100264 ,
-0.08265692, 0.03992791],
[-0.17050642, 0.40016383, 0.0793944 , ..., -0.01002544,
-0.08265692, 0.0399267 ],
...,
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ],
[ 0.03709371, -0.03564092, 0.02344972, ..., 0. ,
0. , 0. ]], dtype=float32), :>,}, children='42->64_0e': input='inputs': Array([[[-0.17050648],
[ 0.40016395],
[ 0.07939404],
...,
[-0.19592644],
[-0.14175692],
[ 0.00325032]],
[[-0.17050642],
[ 0.4001637 ],
[ 0.07939415],
...,
[-0.19592634],
[-0.14175695],
[ 0.00325033]],
[[-0.17050642],
[ 0.40016383],
[ 0.0793944 ],
...,
[-0.19592628],
[-0.14175674],
[ 0.00325033]],
...,
[[ 0.03709371],
[-0.03564092],
[ 0.02344972],
...,
[-0.07406045],
[-0.04630058],
[ 0.2027607 ]],
[[ 0.03709371],
[-0.03564092],
[ 0.02344972],
...,
[-0.07406045],
[-0.04630058],
[ 0.2027607 ]],
[[ 0.03709371],
[-0.03564092],
[ 0.02344972],
...,
[-0.07406045],
[-0.04630058],
[ 0.2027607 ]]], dtype=float32):>,}, params=#=2688, kernel=Array([[ 0.24131659, 0.25181314, 0.13607562, ..., -0.05717197,
-0.00191364, -0.03126345],
[-0.04217541, 0.15758811, -0.18064885, ..., -0.29436707,
-0.02908733, -0.19422905],
[ 0.07417157, 0.02106231, -0.11033244, ..., 0.1088527 ,
0.3471155 , -0.11355939],
...,
[ 0.0243844 , -0.09121892, -0.00826554, ..., -0.08200238,
-0.13456352, 0.07375324],
[-0.32237953, 0.13958512, -0.02433566, ..., 0.09567309,
-0.00787227, 0.03640235],
[-0.10342737, 0.19566657, -0.09860327, ..., -0.24386746,
-0.12877654, 0.13592875]], dtype=float32):>,)>, output=Array([[[ 0.02576888, -0.11656043, -0.13248274, ..., 0.2109735 ,
0.32787305, -0.17879638]],
[[ 0.0257688 , -0.11656041, -0.13248275, ..., 0.2109735 ,
0.32787302, -0.17879641]],
[[ 0.02576882, -0.1165605 , -0.13248266, ..., 0.21097359,
0.3278732 , -0.17879656]],
...,
[[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]],
[[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]],
[[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]]], dtype=float32):>,)>,}, output=64x0e
[[ 0.02576888 -0.11656043 -0.13248274 ... 0.2109735 0.32787305
-0.17879638]
[ 0.0257688 -0.11656041 -0.13248275 ... 0.2109735 0.32787302
-0.17879641]
[ 0.02576882 -0.1165605 -0.13248266 ... 0.21097359 0.3278732
-0.17879656]
...
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]]key=0,): Array([[ 0.02576888, -0.11656043, -0.13248274, ..., 0.2109735 ,
0.32787305, -0.17879638],
[ 0.0257688 , -0.11656041, -0.13248275, ..., 0.2109735 ,
0.32787302, -0.17879641],
[ 0.02576882, -0.1165605 , -0.13248266, ..., 0.21097359,
0.3278732 , -0.17879656],
...,
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]], dtype=float32), :>,)>, 'readout': input='x': 64x0e
[[ 1.3257432 0.26495704 -0.6921134 ... 2.8093698 0.58116055
0.40569487]
[ 1.325743 0.26495636 -0.69211257 ... 2.8093696 0.5811596
0.40569556]
[ 1.325743 0.26495516 -0.6921133 ... 2.8093698 0.5811583
0.4056959 ]
...
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]]key=0,): Array([[ 1.3257432 , 0.26495704, -0.6921134 , ..., 2.8093698 ,
0.58116055, 0.40569487],
[ 1.325743 , 0.26495636, -0.69211257, ..., 2.8093696 ,
0.5811596 , 0.40569556],
[ 1.325743 , 0.26495516, -0.6921133 , ..., 2.8093698 ,
0.5811583 , 0.4056959 ],
...,
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]], dtype=float32), :>,}, children='Linear_0': input='weights_or_input': 64x0e
[[ 1.3257432 0.26495704 -0.6921134 ... 2.8093698 0.58116055
0.40569487]
[ 1.325743 0.26495636 -0.69211257 ... 2.8093696 0.5811596
0.40569556]
[ 1.325743 0.26495516 -0.6921133 ... 2.8093698 0.5811583
0.4056959 ]
...
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]]key=0,): Array([[ 1.3257432 , 0.26495704, -0.6921134 , ..., 2.8093698 ,
0.58116055, 0.40569487],
[ 1.325743 , 0.26495636, -0.69211257, ..., 2.8093696 ,
0.5811596 , 0.40569556],
[ 1.325743 , 0.26495516, -0.6921133 , ..., 2.8093698 ,
0.5811583 , 0.4056959 ],
...,
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]], dtype=float32), :>,}, params=#=4096, w[0,0] 64x0e,64x0e=Array([[ 0.10391267, 1.024953 , -0.40757507, ..., 0.0924962 ,
-2.2268698 , -0.55993104],
[ 0.31792814, 0.45917 , 0.36621046, ..., -0.5914046 ,
-0.06809231, 0.08776443],
[ 0.02763815, -0.3450065 , -0.14293152, ..., 0.15202282,
0.5701436 , 1.4969685 ],
...,
[-0.19278422, -0.15939087, 1.5173362 , ..., -1.0882547 ,
0.5597854 , -0.46402195],
[-0.7843005 , 1.2282125 , 1.1504925 , ..., -0.19969457,
0.45321345, -1.0307884 ],
[ 0.5504516 , -1.1864579 , -1.1046896 , ..., 2.8936856 ,
0.6529835 , 1.652954 ]], dtype=float32):>,)>, output=64x0e
[[ 0.55306005 2.115754 0.9693942 ... -0.71392035 0.5316986
-1.0461974 ]
[ 0.5530611 2.115753 0.96939456 ... -0.71392107 0.5316999
-1.046198 ]
[ 0.5530625 2.1157532 0.9693941 ... -0.7139209 0.5316998
-1.0461979 ]
...
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]key=0,): Array([[ 0.55306005, 2.115754 , 0.9693942 , ..., -0.71392035,
0.5316986 , -1.0461974 ],
[ 0.5530611 , 2.115753 , 0.96939456, ..., -0.71392107,
0.5316999 , -1.046198 ],
[ 0.5530625 , 2.1157532 , 0.9693941 , ..., -0.7139209 ,
0.5316998 , -1.0461979 ],
...,
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]], dtype=float32), :>,)>,}, output=64x0e
[[ 0.55306005 2.115754 0.9693942 ... -0.71392035 0.5316986
-1.0461974 ]
[ 0.5530611 2.115753 0.96939456 ... -0.71392107 0.5316999
-1.046198 ]
[ 0.5530625 2.1157532 0.9693941 ... -0.7139209 0.5316998
-1.0461979 ]
...
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]key=0,): Array([[ 0.55306005, 2.115754 , 0.9693942 , ..., -0.71392035,
0.5316986 , -1.0461974 ],
[ 0.5530611 , 2.115753 , 0.96939456, ..., -0.71392107,
0.5316999 , -1.046198 ],
[ 0.5530625 , 2.1157532 , 0.9693941 , ..., -0.7139209 ,
0.5316998 , -1.0461979 ],
...,
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]], dtype=float32), :>,)>,}, output=64x0e
[[ 1.3257432 0.26495704 -0.6921134 ... 2.8093698 0.58116055
0.40569487]
[ 1.325743 0.26495636 -0.69211257 ... 2.8093696 0.5811596
0.40569556]
[ 1.325743 0.26495516 -0.6921133 ... 2.8093698 0.5811583
0.4056959 ]
...
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]
[ 0.15825638 0.01527765 0.10709445 ... -0.09951308 -0.19287935
0.03289847]]key=0,): Array([[ 1.3257432 , 0.26495704, -0.6921134 , ..., 2.8093698 ,
0.58116055, 0.40569487],
[ 1.325743 , 0.26495636, -0.69211257, ..., 2.8093696 ,
0.5811596 , 0.40569556],
[ 1.325743 , 0.26495516, -0.6921133 , ..., 2.8093698 ,
0.5811583 , 0.4056959 ],
...,
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847],
[ 0.15825638, 0.01527765, 0.10709445, ..., -0.09951308,
-0.19287935, 0.03289847]], dtype=float32), :>, 64x0e
[[ 0.55306005 2.115754 0.9693942 ... -0.71392035 0.5316986
-1.0461974 ]
[ 0.5530611 2.115753 0.96939456 ... -0.71392107 0.5316999
-1.046198 ]
[ 0.5530625 2.1157532 0.9693941 ... -0.7139209 0.5316998
-1.0461979 ]
...
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]
[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]key=0,): Array([[ 0.55306005, 2.115754 , 0.9693942 , ..., -0.71392035,
0.5316986 , -1.0461974 ],
[ 0.5530611 , 2.115753 , 0.96939456, ..., -0.71392107,
0.5316999 , -1.046198 ],
[ 0.5530625 , 2.1157532 , 0.9693941 , ..., -0.7139209 ,
0.5316998 , -1.0461979 ],
...,
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]], dtype=float32), :>,),)>,}, output=64x0e
[[[ 0.55306005 2.115754 0.9693942 ... -0.71392035 0.5316986
-1.0461974 ]]
[[ 0.5530611 2.115753 0.96939456 ... -0.71392107 0.5316999
-1.046198 ]]
[[ 0.5530625 2.1157532 0.9693941 ... -0.7139209 0.5316998
-1.0461979 ]]
...
[[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]
[[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]
[[-0.24056832 -0.08530777 -0.06109358 ... 0.23549592 -0.25863734
-0.1854789 ]]]key=0,): Array([[[ 0.55306005, 2.115754 , 0.9693942 , ..., -0.71392035,
0.5316986 , -1.0461974 ]],
[[ 0.5530611 , 2.115753 , 0.96939456, ..., -0.71392107,
0.5316999 , -1.046198 ]],
[[ 0.5530625 , 2.1157532 , 0.9693941 , ..., -0.7139209 ,
0.5316998 , -1.0461979 ]],
...,
[[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]],
[[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]],
[[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]]], dtype=float32), :>,)>, 'norm': input='x': Array([[ 0.55306005, 2.115754 , 0.9693942 , ..., -0.71392035,
0.5316986 , -1.0461974 ],
[ 0.5530611 , 2.115753 , 0.96939456, ..., -0.71392107,
0.5316999 , -1.046198 ],
[ 0.5530625 , 2.1157532 , 0.9693941 , ..., -0.7139209 ,
0.5316998 , -1.0461979 ],
...,
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ],
[-0.24056832, -0.08530777, -0.06109358, ..., 0.23549592,
-0.25863734, -0.1854789 ]], dtype=float32):>,}, params=#=128, scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32):>, bias=Array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32):>,)>, output=Array([[ 0.67563576, 2.1714454 , 1.0741506 , ..., -0.5371171 ,
0.65518856, -0.8551725 ],
[ 0.67563677, 2.1714444 , 1.0741508 , ..., -0.5371178 ,
0.6551898 , -0.85517305],
[ 0.6756381 , 2.1714447 , 1.0741504 , ..., -0.53711766,
0.6551897 , -0.855173 ],
...,
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ]], dtype=float32):>,)>, 'head': input='x': Array([[ 0.67563576, 2.1714454 , 1.0741506 , ..., -0.5371171 ,
0.65518856, -0.8551725 ],
[ 0.67563677, 2.1714444 , 1.0741508 , ..., -0.5371178 ,
0.6551898 , -0.85517305],
[ 0.6756381 , 2.1714447 , 1.0741504 , ..., -0.53711766,
0.6551897 , -0.855173 ],
...,
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ]], dtype=float32):>,}, children='Dense_0': input='inputs': Array([[ 0.67563576, 2.1714454 , 1.0741506 , ..., -0.5371171 ,
0.65518856, -0.8551725 ],
[ 0.67563677, 2.1714444 , 1.0741508 , ..., -0.5371178 ,
0.6551898 , -0.85517305],
[ 0.6756381 , 2.1714447 , 1.0741504 , ..., -0.53711766,
0.6551897 , -0.855173 ],
...,
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ],
[-1.983305 , -0.6997215 , -0.49953592, ..., 1.9524546 ,
-2.1326869 , -1.5278648 ]], dtype=float32):>,}, params=#=2048, kernel=Array([[-0.15120558, 0.04598498, -0.1182805 , ..., 0.19351044,
-0.15725753, 0.11815574],
[-0.13509426, -0.02575719, -0.07142678, ..., 0.02042856,
-0.06408975, -0.06645787],
[ 0.0462003 , 0.05232048, 0.1236474 , ..., -0.10480619,
-0.03226347, -0.09103674],
...,
[ 0.2867331 , 0.04106171, -0.06965617, ..., -0.0973758 ,
0.1761544 , 0.24882723],
[-0.28845236, 0.01888539, 0.21720028, ..., -0.05952445,
-0.05499991, 0.11649268],
[ 0.01211685, -0.2016966 , -0.04587143, ..., -0.26204085,
0.06534766, 0.05331267]], dtype=float32):>,)>, output=Array([[-0.4161812 , -0.6380119 , 0.23795176, ..., 0.44758654,
0.55685276, -0.42165393],
[-0.41618097, -0.6380104 , 0.23795158, ..., 0.44758666,
0.55685264, -0.42165384],
[-0.41618335, -0.6380107 , 0.23795247, ..., 0.447586 ,
0.55685294, -0.4216536 ],
...,
[ 1.9206522 , 0.8000797 , -0.6356598 , ..., 0.98179626,
0.11009294, -0.82140815],
[ 1.9206522 , 0.8000797 , -0.6356598 , ..., 0.98179626,
0.11009294, -0.82140815],
[ 1.9206522 , 0.8000797 , -0.6356598 , ..., 0.98179626,
0.11009294, -0.82140815]], dtype=float32):>,)>, 'Dropout_0': input='inputs': Array([[-0.16540328, -0.22055823, 0.13306473, ..., 0.27305698,
0.35400468, -0.167026 ],
[-0.16540323, -0.22055793, 0.1330646 , ..., 0.27305707,
0.35400456, -0.16702597],
[-0.16540392, -0.22055797, 0.13306516, ..., 0.27305663,
0.3540048 , -0.1670259 ],
...,
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ]], dtype=float32):>,}, output=Array([[-0.16540328, -0.22055823, 0.13306473, ..., 0.27305698,
0.35400468, -0.167026 ],
[-0.16540323, -0.22055793, 0.1330646 , ..., 0.27305707,
0.35400456, -0.16702597],
[-0.16540392, -0.22055797, 0.13306516, ..., 0.27305663,
0.3540048 , -0.1670259 ],
...,
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ]], dtype=float32):>,)>, 'LayerNorm_0': input='x': Array([[-0.16540328, -0.22055823, 0.13306473, ..., 0.27305698,
0.35400468, -0.167026 ],
[-0.16540323, -0.22055793, 0.1330646 , ..., 0.27305707,
0.35400456, -0.16702597],
[-0.16540392, -0.22055797, 0.13306516, ..., 0.27305663,
0.3540048 , -0.1670259 ],
...,
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ],
[ 1.6752143 , 0.5520482 , -0.22008343, ..., 0.71422195,
0.05807352, -0.2509113 ]], dtype=float32):>,}, params=#=32, scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32):>,)>, output=Array([[-0.7410716 , -0.8437338 , -0.18552074, ..., 0.07505263,
0.22572392, -0.74409205],
[-0.7410714 , -0.8437331 , -0.18552092, ..., 0.07505284,
0.22572373, -0.74409187],
[-0.7410729 , -0.8437335 , -0.18551984, ..., 0.07505216,
0.2257244 , -0.7440919 ],
...,
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ],
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ],
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ]], dtype=float32):>,)>, 'Dense_1': input='inputs': Array([[-0.7410716 , -0.8437338 , -0.18552074, ..., 0.07505263,
0.22572392, -0.74409205],
[-0.7410714 , -0.8437331 , -0.18552092, ..., 0.07505284,
0.22572373, -0.74409187],
[-0.7410729 , -0.8437335 , -0.18551984, ..., 0.07505216,
0.2257244 , -0.7440919 ],
...,
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ],
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ],
[ 2.5978398 , 0.4883031 , -0.96191883, ..., 0.7928987 ,
-0.4394829 , -1.0198199 ]], dtype=float32):>,}, params=#=32, kernel=Array([[ 0.15032198],
[ 0.02310337],
[ 0.05576847],
[-0.29912832],
[ 0.27433982],
[ 0.47631413],
[ 0.05964498],
[ 0.40425548],
[ 0.20762067],
[ 0.06157257],
[ 0.05725569],
[ 0.1319021 ],
[ 0.03838015],
[ 0.3880695 ],
[-0.42079282],
[ 0.2898683 ],
[ 0.205256 ],
[ 0.30502445],
[-0.10051804],
[-0.0186508 ],
[-0.04129213],
[ 0.22162631],
[-0.02105439],
[ 0.5244865 ],
[ 0.27792183],
[ 0.17122482],
[-0.16301072],
[-0.00705524],
[ 0.20203777],
[-0.01598651],
[-0.08380979],
[-0.29766732]], dtype=float32):>,)>, output=Array([[-0.48011217],
[-0.4801123 ],
[-0.48011047],
...,
[ 0.09121275],
[ 0.09121275],
[ 0.09121275]], dtype=float32):>,)>,}, output=Array([[-0.48011217],
[-0.4801123 ],
[-0.48011047],
...,
[ 0.09121275],
[ 0.09121275],
[ 0.09121275]], dtype=float32):>,)>, 'rescale': input='energies': Array([-0.48011217, -0.4801123 , -0.48011047, ..., 0.09121275,
0.09121275, 0.09121275], dtype=float32):>, 'node_species': Array([14, 14, 14, ..., 0, 0, 0], dtype=int16):>,}, params=#=178, shift=Array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0.], dtype=float32):>, scale=Array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1.], dtype=float32):>,)>, output=Array([-0.48011217, -0.4801123 , -0.48011047, ..., 0.09121275,
0.09121275, 0.09121275], dtype=float32):>,)>, 'node2graph_reduction': input='data': Array([-0.48011217, -0.4801123 , -0.48011047, ..., 0.09121275,
0.09121275, 0.09121275], dtype=float32):>, 'segments': Array([ 0, 0, 0, ..., 28, 28, 28], dtype=int32):>, 'num_segments': 32,}, output=Array([-3.5796541e-01, 1.4956104e+00, 3.4306303e-04, 9.5203966e-01,
-1.2770801e+00, -1.7660536e+00, -7.7776408e-01, -1.4580652e+00,
6.3796794e-01, -5.5803972e-01, -1.5454605e+00, -8.7018156e-01,
-1.4528267e+00, 4.4214651e-01, -5.0913250e-01, -6.8798405e-01,
1.9180411e-01, -5.2204072e-01, -4.9298960e-01, 7.9337984e-01,
-5.6942111e-01, -1.9111191e-01, -8.8395828e-01, 8.3204418e-01,
-2.9291064e-01, 1.3669008e+00, -2.4115901e+00, -1.2340176e-02,
8.8633783e-02, 0.0000000e+00, 0.0000000e+00, 0.0000000e+00], dtype=float32):>,)>,}, output=Array([[ -6.534635 ],
[ -3.2923043],
[ -5.9078712],
[ -4.2431345],
[ -8.142378 ],
[ -8.997705 ],
[ -7.2689595],
[ -8.458962 ],
[ -4.7925186],
[ -6.884611 ],
[ -8.611837 ],
[ -7.430619 ],
[ -8.4498 ],
[ -5.135055 ],
[ -6.799061 ],
[ -7.1119137],
[ -5.572962 ],
[ -6.8216405],
[ -6.7708235],
[ -4.520667 ],
[ -6.9045196],
[ -6.2427697],
[ -7.4547176],
[ -4.4530344],
[ -6.4208393],
[ -3.517447 ],
[-10.126898 ],
[ -5.930057 ],
[ -5.7534304],
[ -5.908471 ],
[ -5.908471 ],
[ -5.908471 ]], dtype=float32):>,)>